Skip to content

Commit

Permalink
Disable py3.12 wheel builds for ROCm (#5301)
Browse files Browse the repository at this point in the history
Reference:
pytorch/pytorch#127968 (review)

Triton commit bump PR pytorch/pytorch#125396
reverted due to missing llnl-hatchet dependency for triton. Workaround
is to disable py3.12 wheel build jobs for ROCm on PyTorch CI until
llnl-hatchet publishes py3.12 wheels on
[PyPI](https://pypi.org/project/llnl-hatchet/#files)
  • Loading branch information
jithunnair-amd authored Jun 6, 2024
1 parent 20eabd8 commit e9d8b21
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 2 deletions.
4 changes: 4 additions & 0 deletions tools/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down
Loading

0 comments on commit e9d8b21

Please sign in to comment.