Disable py3.12 wheel builds for ROCm

pytorch · Jun 5, 2024 · e3a8fcb · e3a8fcb
1 parent 28e73d0
commit e3a8fcb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -495,6 +495,10 @@ def generate_wheels_matrix(
     for python_version in python_versions:
         for arch_version in arches:
             gpu_arch_type = arch_type(arch_version)
+            # Disable py3.12 builds for ROCm because of triton dependency
+            # on llnl-hatchet, which doesn't have py3.12 wheels available
+            if gpu_arch_type == ROCM and python_version == "3.12":
+                continue
             gpu_arch_version = (
                 ""
                 if arch_version in [CPU, CPU_AARCH64]