Remove py312 and rocm restriction (#5508)

Revert of #5301 since not required anymore. This dependency is made optional
pytorch · Jul 26, 2024 · 5e1498d · 5e1498d
1 parent e7848f5
commit 5e1498d
Show file tree

Hide file tree

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

diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json