We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4594274 commit 129c488Copy full SHA for 129c488
setup.py
@@ -112,7 +112,7 @@ def append_nvcc_threads(nvcc_extra_args):
112
if bare_metal_version >= Version("11.8"):
113
cc_flag.append("-gencode")
114
cc_flag.append("arch=compute_90,code=sm_90")
115
- if bare_metal_version >= Version("12.8.1"):
+ if bare_metal_version >= Version("12.8"):
116
117
cc_flag.append("arch=compute_100,code=sm_100")
118
0 commit comments