removing build target sm_70 from bnp (#683)

b66ffc1d · jjsjann123 · ptrblck · b5a7c5f9 · b66ffc1d
Commit b66ffc1d authored Jan 20, 2020 by jjsjann123 Committed by ptrblck Jan 20, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

setup.py setup.py +1 -3

No files found.
--- a/setup.py
+++ b/setup.py
@@ -153,9 +153,7 @@ if "--bnp" in sys.argv:
                                              'nvcc':['-DCUDA_HAS_FP16=1',
                                                      '-D__CUDA_NO_HALF_OPERATORS__',
                                                      '-D__CUDA_NO_HALF_CONVERSIONS__',
-                                                      '-D__CUDA_NO_HALF2_OPERATORS__',
-                                                      '-gencode',
-                                                      'arch=compute_70,code=sm_70'] + version_dependent_macros}))
+                                                      '-D__CUDA_NO_HALF2_OPERATORS__'] + version_dependent_macros}))

 if "--xentropy" in sys.argv:
    from torch.utils.cpp_extension import CUDAExtension