diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index e760d94494..74799cdb41 100644 --- 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]