Skip to content

Commit 81e1bc9

Browse files
committed
chore: add target sm121 only if cuda>=13
1 parent 896fb86 commit 81e1bc9

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

setup.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ def get_sm_targets() -> list[str]:
3939
raise Exception("nvcc not found")
4040

4141
support_sm120 = packaging_version.parse(nvcc_version) >= packaging_version.parse("12.8")
42+
support_sm121 = packaging_version.parse(nvcc_version) >= packaging_version.parse("13.0")
4243

4344
install_mode = os.getenv("NUNCHAKU_INSTALL_MODE", "FAST")
4445
if install_mode == "FAST":
@@ -55,7 +56,9 @@ def get_sm_targets() -> list[str]:
5556
assert install_mode == "ALL"
5657
ret = ["75", "80", "86", "89"]
5758
if support_sm120:
58-
ret.extend(["120a", "121a"])
59+
ret.extend(["120a"])
60+
if support_sm121:
61+
ret.extend(["121a"])
5962
return ret
6063

6164

0 commit comments

Comments
 (0)