add cu126 torchvision and torchaudio aarch64 build

pytorch · Dec 16, 2024 · cf7d0f8 · cf7d0f8
1 parent 28ccdef
commit cf7d0f8
Show file tree

Hide file tree

Showing 11 changed files with 11 additions and 2,003 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -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/assets/build_matrix_linux_conda_cuda.json b/tools/tests/assets/build_matrix_linux_conda_cuda.json
@@ -1,3 +1 @@
-{
-  "include": []
-}
+{"include": []}