From 272066566f31f5534a18554f9e07d8ce0c8aab72 Mon Sep 17 00:00:00 2001 From: YC Yen-Ching Tseng Date: Fri, 29 May 2026 11:34:45 +0800 Subject: [PATCH] [AMD] Pin compressed-tensors<0.16.0 for srt_hip (fixes ROCm 7.2 nightly build) (#26591) --- python/pyproject_other.toml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml index 52f04854d..86ccfe910 100755 --- a/python/pyproject_other.toml +++ b/python/pyproject_other.toml @@ -95,6 +95,13 @@ srt_hip = [ "torch", "petit_kernel==0.0.2", "wave-lang==3.8.2", + # HOTFIX (2026-05-28): compressed-tensors 0.16.0 added `torch>=2.10.0`, + # which forces pip off the ROCm wheel (torch 2.9.1+rocm7.2) and silently + # swaps it for the PyPI default `torch 2.12.0+cu130`, breaking the + # downstream HIP build of fast-hadamard-transform with a NameError on + # `bare_metal_version`. Pin below 0.16.0 until the ROCm base ships + # torch>=2.10. + "compressed-tensors<0.16.0", ] diffusion_hip = [