Skip to content

Commit 4a8a2a2

Browse files
authored
add B200 support (#726)
Signed-off-by: Mayank Mishra <[email protected]>
1 parent 2e16fc3 commit 4a8a2a2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

setup.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,9 @@ def append_nvcc_threads(nvcc_extra_args):
188188
if bare_metal_version >= Version("11.8"):
189189
cc_flag.append("-gencode")
190190
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")
191194

192195

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

0 commit comments

Comments
 (0)