From 705b7736b49320accb1bc40ade1e317bbed34677 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Wed, 5 Jun 2024 17:02:18 +0000 Subject: [PATCH] Disable py3.12 wheel builds for ROCm --- tools/scripts/generate_binary_build_matrix.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 90b9bf160c..698badf07d 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -500,6 +500,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]