Add changes for binary_build_matrix

pytorch · Dec 16, 2024 · bf1d488 · bf1d488
1 parent 6a15e32
commit bf1d488
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -163,7 +163,7 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
         XPU: "pytorch/manylinux2_28-builder:xpu",
         # TODO: Migrate CUDA_AARCH64 image to manylinux2_28_aarch64-builder:cuda12.4
         CPU_AARCH64: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64",
-        CUDA_AARCH64: "pytorch/manylinuxaarch64-builder:cuda12.4",
+        CUDA_AARCH64: "pytorch/manylinuxaarch64-builder:cuda12.6",
     }
     LIBTORCH_CONTAINER_IMAGES = {
         **{
@@ -191,7 +191,7 @@ def translate_desired_cuda(gpu_arch_type: str, gpu_arch_version: str) -> str:
     return {
         CPU: "cpu",
         CPU_AARCH64: CPU,
-        CUDA_AARCH64: "cu124",
+        CUDA_AARCH64: "cu126",
         CUDA: f"cu{gpu_arch_version.replace('.', '')}",
         ROCM: f"rocm{gpu_arch_version}",
         XPU: "xpu",

diff --git a/tools/tests/update_test_assets.sh b/tools/tests/update_test_assets.sh
@@ -1,12 +1,12 @@
 
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_cuda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-rocm disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_cuda_norocm.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-cpu disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_nocpu.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-cpu disable --with-rocm disable  --with-xpu enable > ../tests/assets/build_matrix_linux_wheel_xpu.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --package-type conda > ../tests/assets/build_matrix_linux_conda_cuda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --package-type conda > ../tests/assets/build_matrix_linux_conda_cuda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="macos" --with-cuda disable --with-rocm disable > ../tests/assets/build_matrix_macos_wheel.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="macos" --package-type conda  > ../tests/assets/build_matrix_macos_conda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="windows" --package-type conda > ../tests/assets/build_matrix_windows_conda_cuda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="windows" > ../tests/assets/build_matrix_windows_wheel_cuda.json
-python3 ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-rocm disable --with-cuda disable --operating-system="windows" > ../tests/assets/build_matrix_windows_wheel_xpu.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_cuda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-rocm disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_cuda_norocm.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-cpu disable --with-xpu disable > ../tests/assets/build_matrix_linux_wheel_nocpu.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-cpu disable --with-rocm disable  --with-xpu enable > ../tests/assets/build_matrix_linux_wheel_xpu.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --package-type conda > ../tests/assets/build_matrix_linux_conda_cuda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --package-type conda > ../tests/assets/build_matrix_linux_conda_cuda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="macos" --with-cuda disable --with-rocm disable > ../tests/assets/build_matrix_macos_wheel.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="macos" --package-type conda  > ../tests/assets/build_matrix_macos_conda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="windows" --package-type conda > ../tests/assets/build_matrix_windows_conda_cuda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --operating-system="windows" > ../tests/assets/build_matrix_windows_wheel_cuda.json
+python ../scripts/generate_binary_build_matrix.py --build-python-only disable --with-rocm disable --with-cuda disable --operating-system="windows" > ../tests/assets/build_matrix_windows_wheel_xpu.json