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