From bfe1001ab6d59d7da1f70a3bc107cebaff91f925 Mon Sep 17 00:00:00 2001 From: TJian Date: Thu, 11 Jun 2026 01:41:15 +0800 Subject: [PATCH] [Bugfix] [DSV4] [ROCm] Pin apache-tvm-ffi version to `0.1.10` (#45169) Signed-off-by: tjtanaa --- requirements/rocm.txt | 3 ++- requirements/test/rocm.txt | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 4ca7073830..5179f6ee8d 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -23,6 +23,7 @@ timm>=1.0.17 # To be consistent with test_quark.py amd-quark>=0.8.99 tilelang==0.1.10 - +# Required apache-tvm-ffi matching tilelang version +apache-tvm-ffi==0.1.10 # Required for faster safetensors model loading fastsafetensors >= 0.3.2 diff --git a/requirements/test/rocm.txt b/requirements/test/rocm.txt index eced711711..ce18ce456c 100644 --- a/requirements/test/rocm.txt +++ b/requirements/test/rocm.txt @@ -44,6 +44,7 @@ anyio==4.13.0 # watchfiles apache-tvm-ffi==0.1.10 # via + # -c requirements/rocm.txt # tilelang # xgrammar arctic-inference==0.1.1