removing build target sm_70 from bnp (NVIDIA#683)

ingeniousfrog · Jan 21, 2020 · b66ffc1 · b66ffc1
1 parent b5a7c5f
commit b66ffc1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -153,9 +153,7 @@ def check_cuda_torch_binary_vs_bare_metal(cuda_dir):
                                               '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