[Dependency] Upgrade FlashInfer to 0.7.1 (#60948)

Signed-off-by: wzhao18 <wzhao18.sz@gmail.com>
Co-authored-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
Wei Zhao
2026-10-14 12:31:08 -07:00
committed by GitHub
co-authored by Jee Jee Li
parent 9b5c6707a3
commit 4975fda882
2 changed files with 2 additions and 1 deletions
+1 -1
View File
@@ -10,7 +10,7 @@ torchaudio==2.11.0
torchvision==0.29.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
torchcodec >= 0.14
PyNvVideoCodec==2.0.4
flashinfer-python==0.7.0.post1
flashinfer-python==0.7.1
apache-tvm-ffi==0.1.11
tilelang==0.1.12
nvidia-cudnn-frontend>=1.19.1
+1
View File
@@ -456,6 +456,7 @@ def _flashinfer_kda_prefill(
beta_is_logit=True,
seq_order=seq_order,
prefill_workspace=prefill_workspace,
backend="cute-dsl",
)
return output, initial_state