We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 36b2e38 commit 971ac9dCopy full SHA for 971ac9d
tools/scripts/generate_binary_build_matrix.py
@@ -443,17 +443,14 @@ def generate_wheels_matrix(
443
arches = []
444
445
if with_cpu == ENABLE:
446
- arches += [CPU]
447
-
448
- if os == LINUX_AARCH64:
449
- # Only want the one arch as the CPU type is different and
450
- # uses different build/test scripts
451
- arches = [CPU_AARCH64] + CUDA_AARCH64_ARCHES
+ arches += [CPU_AARCH64] if os == LINUX_AARCH64 else [CPU]
452
453
if with_cuda == ENABLE:
454
upload_to_base_bucket = "no"
455
if os in (LINUX, WINDOWS):
456
arches += CUDA_ARCHES
+ elif os == LINUX_AARCH64:
+ arches += CUDA_AARCH64_ARCHES
457
458
if with_rocm == ENABLE and os == LINUX:
459
arches += ROCM_ARCHES
0 commit comments