Update flashinfer to 0.3.1 for B300 support (#10087)

Signed-off-by: Hao Lu <14827759+hlu1@users.noreply.github.com>
This commit is contained in:
hlu1
2025-09-05 13:41:01 -07:00
committed by GitHub
parent e678cc717d
commit 2985090084
2 changed files with 3 additions and 3 deletions

4
python/pyproject.toml Normal file → Executable file
View File

@@ -63,7 +63,7 @@ srt = [
"torchaudio==2.8.0",
"torchvision",
"cuda-python",
"flashinfer_python==0.3.0",
"flashinfer_python==0.3.1",
]
blackwell = [
@@ -73,7 +73,7 @@ blackwell = [
"torchaudio==2.8.0",
"torchvision",
"cuda-python",
"flashinfer_python==0.3.0",
"flashinfer_python==0.3.1",
]
# HIP (Heterogeneous-computing Interface for Portability) for AMD

View File

@@ -673,7 +673,7 @@ def _set_envs_and_config(server_args: ServerArgs):
if server_args.attention_backend == "flashinfer":
assert_pkg_version(
"flashinfer_python",
"0.3.0",
"0.3.1",
"Please uninstall the old version and "
"reinstall the latest version by following the instructions "
"at https://docs.flashinfer.ai/installation.html.",