Skip to content

Commit e41ebd9

Browse files
Breakthroughclaude
andcommitted
[build] Update CUDA arch for CUDA 13.x compatibility
CUDA 13.x dropped support for compute capability < 7.0. Updated to support Volta (7.0) through Hopper (9.0). Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent 5bdb5ff commit e41ebd9

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

setup.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@ def main():
2323
build_rolling = get_build_env_var_by_name("rolling")
2424

2525
# TODO(@Breakthrough): What architectures should we ship?
26+
# TODO(@Breakthrough): Create a separate builder/shard using CUDA 12.x for older GPU
27+
# support (Maxwell/Pascal: 5.0, 5.2, 6.0, 6.1). This would allow shipping two versions
28+
# of the build artifacts with varying backwards compatibility.
2629
# fermi = "2.0"
2730
# kepler = "3.0;3.5;3.7"
2831
# maxwell = "5.0;5.2"
@@ -33,8 +36,9 @@ def main():
3336
# lovelace = "8.9"
3437
# hopper = "9.0"
3538
# blackwell = "10.0;12.0"
36-
cuda_arch_bin = "5.0;5.2;6.0;6.1;7.0;7.5,8.0"
37-
cuda_arch_ptx = "8.0"
39+
# CUDA 13.x dropped support for compute capability < 7.5 (Turing)
40+
cuda_arch_bin = "7.5;8.0;8.6;8.9;9.0"
41+
cuda_arch_ptx = "9.0"
3842

3943
# NOTE: since 2.3.0 numpy upgraded from manylinux2014 to manylinux_2_28
4044
# see https://numpy.org/doc/stable/release/2.3.0-notes.html#numpy-2-3-0-release-notes

0 commit comments

Comments
 (0)