Skip to content

Commit 2aca540

Browse files
Add additional CUDA architecture options for Blackwell series
1 parent 457b73e commit 2aca540

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

CMakeLists.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,12 @@ set_property(CACHE CUDA_ARCH PROPERTY STRINGS
2222
"80 - Ampere (A100)"
2323
"86 - Ampere (RTX 30xx)"
2424
"89 - Ada Lovelace (RTX 40xx)"
25-
"90 - Hopper (H100)"
25+
"90 - Hopper (H100)"
26+
"100 - Blackwell (GB200, B200)"
27+
"103 - Blackwell (GB300, B300)"
28+
"110 - Blackwell Jetson (T5000, T4000)"
29+
"120 - Blackwell (RTX 50xx, RTX PRO)"
30+
"121 - Blackwell (DGX Spark GB10)"
2631
)
2732

2833
# Extract just the number from the selection (digits before the " - " label)

0 commit comments

Comments
 (0)