Skip to content

Commit

Permalink
only remove 12.1
Browse files Browse the repository at this point in the history
  • Loading branch information
tinglvv committed Nov 12, 2024
1 parent 0d757ae commit 422c35b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tools/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,9 @@
"release": ["3.9", "3.10", "3.11", "3.12"],
}
CUDA_ARCHES_DICT = {
"nightly": ["11.8", "12.4", "12.6"],
"test": ["11.8", "12.4", "12.6"],
"release": ["11.8", "12.4", "12.6"],
"nightly": ["11.8", "12.4"],
"test": ["11.8", "12.1", "12.4"],
"release": ["11.8", "12.1", "12.4"],
}
ROCM_ARCHES_DICT = {
"nightly": ["6.1", "6.2"],
Expand All @@ -37,8 +37,8 @@

CUDA_CUDDN_VERSIONS = {
"11.8": {"cuda": "11.8.0", "cudnn": "9"},
"12.1": {"cuda": "12.1.1", "cudnn": "9"},
"12.4": {"cuda": "12.4.1", "cudnn": "9"},
"12.6": {"cuda": "12.6.2", "cudnn": "9"},
}

PACKAGE_TYPES = ["wheel", "conda", "libtorch"]
Expand Down

0 comments on commit 422c35b

Please sign in to comment.