diff --git a/setup.py b/setup.py index 7d89db4..142addc 100644 --- a/setup.py +++ b/setup.py @@ -101,10 +101,18 @@ def append_nvcc_threads(nvcc_extra_args): "Note: make sure nvcc has a supported version by running nvcc -V." ) + cc_flag.append("-gencode") + cc_flag.append("arch=compute_53,code=sm_53") + cc_flag.append("-gencode") + cc_flag.append("arch=compute_62,code=sm_62") cc_flag.append("-gencode") cc_flag.append("arch=compute_70,code=sm_70") cc_flag.append("-gencode") + cc_flag.append("arch=compute_72,code=sm_72") + cc_flag.append("-gencode") cc_flag.append("arch=compute_80,code=sm_80") + cc_flag.append("-gencode") + cc_flag.append("arch=compute_87,code=sm_87") if bare_metal_version >= Version("11.8"): cc_flag.append("-gencode") cc_flag.append("arch=compute_90,code=sm_90")