From 89410b380b2aa58662049e404d94ba79c0eeecb7 Mon Sep 17 00:00:00 2001 From: Bingxu Chen Date: Mon, 1 Jun 2026 17:33:00 +0800 Subject: [PATCH] [AMD] Pin compressed-tensors==0.15.0 to fix ROCm nightly build (#26879) Co-authored-by: Claude Opus 4.8 (1M context) --- python/pyproject_other.toml | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml index 631ecfa31..2e8a49436 100755 --- a/python/pyproject_other.toml +++ b/python/pyproject_other.toml @@ -96,13 +96,10 @@ 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", + # Pin to 0.15.0: 0.16.0 needs torch>=2.10 (incompatible with ROCm torch + # 2.9.1). An open-ended `<0.16.0` made pip backtrack into an unbuildable + # ancient setuptools sdist; an exact pin keeps the resolver converging. + "compressed-tensors==0.15.0", ] diffusion_hip = [