Remove rocm from 3.13 builds

pytorch · Oct 17, 2024 · a114939 · a114939
1 parent 1b4a384
commit a114939
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -530,6 +530,11 @@ def generate_wheels_matrix(
     ret: List[Dict[str, str]] = []
     for python_version in python_versions:
         for arch_version in arches:
+
+            # TODO: Enable Python 3.13 support for ROCM
+            if arch_version in ROCM_ARCHES and python_version == "3.13":
+                continue
+
             gpu_arch_type = arch_type(arch_version)
             gpu_arch_version = (
                 "" if arch_version in [CPU, CPU_AARCH64, XPU] else arch_version