Skip to content

Commit d0ac0de

Browse files
committed
Update os check
Signed-off-by: Huy Do <huydhn@gmail.com>
1 parent 6bf5f52 commit d0ac0de

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tools/scripts/generate_binary_build_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ def initialize_globals(channel: str, os: str, build_python_only: bool) -> None:
157157
CURRENT_VERSION = CURRENT_STABLE_VERSION
158158

159159
CUDA_ARCHES = CUDA_ARCHES_DICT[channel]
160-
if channel != "release" and os == LINUX:
160+
if channel != "release" and (os == LINUX or os == LINUX_AARCH64):
161161
# TODO (huydhn): Only build CUDA 12.9 for Linux. This logic is to be cleaned up
162162
# in 2.10
163163
CUDA_ARCHES.append("12.9")

0 commit comments

Comments
 (0)