We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2e16fc3 commit 4a8a2a2Copy full SHA for 4a8a2a2
setup.py
@@ -188,6 +188,9 @@ def append_nvcc_threads(nvcc_extra_args):
188
if bare_metal_version >= Version("11.8"):
189
cc_flag.append("-gencode")
190
cc_flag.append("arch=compute_90,code=sm_90")
191
+ if bare_metal_version >= Version("12.8"):
192
+ cc_flag.append("-gencode")
193
+ cc_flag.append("arch=compute_100,code=sm_100")
194
195
196
# HACK: The compiler flag -D_GLIBCXX_USE_CXX11_ABI is set to be the same as
0 commit comments