Skip to content

Commit da71f21

Browse files
committed
Don't compile for sm < 75 if nvcc >= 13.0
1 parent e0761ec commit da71f21

1 file changed

Lines changed: 12 additions & 8 deletions

File tree

setup.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -172,25 +172,29 @@ def append_nvcc_threads(nvcc_extra_args):
172172
"Note: make sure nvcc has a supported version by running nvcc -V."
173173
)
174174

175+
if bare_metal_version <= Version("12.9"):
176+
cc_flag.append("-gencode")
177+
cc_flag.append("arch=compute_53,code=sm_53")
178+
cc_flag.append("-gencode")
179+
cc_flag.append("arch=compute_62,code=sm_62")
180+
cc_flag.append("-gencode")
181+
cc_flag.append("arch=compute_70,code=sm_70")
182+
cc_flag.append("-gencode")
183+
cc_flag.append("arch=compute_72,code=sm_72")
175184
cc_flag.append("-gencode")
176-
cc_flag.append("arch=compute_53,code=sm_53")
177-
cc_flag.append("-gencode")
178-
cc_flag.append("arch=compute_62,code=sm_62")
179-
cc_flag.append("-gencode")
180-
cc_flag.append("arch=compute_70,code=sm_70")
181-
cc_flag.append("-gencode")
182-
cc_flag.append("arch=compute_72,code=sm_72")
185+
cc_flag.append("arch=compute_75,code=sm_75")
183186
cc_flag.append("-gencode")
184187
cc_flag.append("arch=compute_80,code=sm_80")
185188
cc_flag.append("-gencode")
186189
cc_flag.append("arch=compute_87,code=sm_87")
187-
188190
if bare_metal_version >= Version("11.8"):
189191
cc_flag.append("-gencode")
190192
cc_flag.append("arch=compute_90,code=sm_90")
191193
if bare_metal_version >= Version("12.8"):
192194
cc_flag.append("-gencode")
193195
cc_flag.append("arch=compute_100,code=sm_100")
196+
cc_flag.append("-gencode")
197+
cc_flag.append("arch=compute_120,code=sm_120")
194198

195199

196200
# HACK: The compiler flag -D_GLIBCXX_USE_CXX11_ABI is set to be the same as

0 commit comments

Comments
 (0)