Skip to content

Commit cc4cf10

Browse files
committed
feat(build): update CUDA architecture support for different versions
1 parent 919951e commit cc4cf10

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

‎.github/workflows/build.yml‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -534,12 +534,17 @@ jobs:
534534
# 61=Pascal (GTX 10), 75=Turing (RTX 20/GTX 16), 86=Ampere (RTX 30), 89=Ada (RTX 40), 120=Blackwell (RTX 50)
535535
$cudaVersion = "${{ matrix.cuda_version }}"
536536
if ($cudaVersion -match "^11\.") {
537-
# CUDA 11.x doesn't support arch 120
537+
# CUDA 11.x: Support Pascal (61) to Ada (89)
538538
$cudaArchs = "61;75;86;89"
539539
$env:CUDAFLAGS = "-allow-unsupported-compiler -D_ALLOW_COMPILER_AND_STL_VERSION_MISMATCH -D_DISABLE_CONSTEXPR_MUTEX_CONSTRUCTOR"
540+
} elseif ($cudaVersion -match "^12\.") {
541+
# CUDA 12.x: Support Pascal (61) to Blackwell (120) (technically 120 needs 12.8+, keeping unified for now or simple)
542+
# Actually 12.x still supports Pascal.
543+
$cudaArchs = "61;75;86;89"
544+
$env:CUDAFLAGS = ""
540545
} else {
541-
# CUDA 12+: Add RTX 50 (Blackwell)
542-
$cudaArchs = "61;75;86;89;120"
546+
# CUDA 13+: Drop Pascal (61). Support Turing (75) to Blackwell (120)
547+
$cudaArchs = "75;86;89;120"
543548
$env:CUDAFLAGS = ""
544549
}
545550
Write-Host "Building for CUDA architectures: $cudaArchs"

0 commit comments

Comments
 (0)