diff --git a/setup.py b/setup.py index 816819dd3..2839cfd88 100644 --- 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