Cherry Pick of #5508 (#5509)

Same change as #5508 for release 2.4
pytorch · Jul 26, 2024 · 281bd19 · 281bd19
1 parent b8f5d7e
commit 281bd19
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -500,10 +500,6 @@ 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]