diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml
index 0b05030df..cebe0abf0 100755
--- a/python/pyproject_other.toml
+++ b/python/pyproject_other.toml
@@ -70,7 +70,7 @@ srt = [
     "torchaudio==2.8.0",
     "torchvision",
     "cuda-python",
-    "flashinfer_python==0.3.1",
+    "flashinfer_python==0.4.0rc1",
 ]
 
 blackwell = [
@@ -80,7 +80,7 @@ blackwell = [
     "torchaudio==2.8.0",
     "torchvision",
     "cuda-python",
-    "flashinfer_python==0.3.1",
+    "flashinfer_python==0.4.0rc1",
     "nvidia-cutlass-dsl==4.2.0",
 ]