diff --git a/python/pyproject.toml b/python/pyproject.toml
index 64915df6b..1cf32215d 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -79,7 +79,7 @@ blackwell = [
 srt_hip = [
     "sglang[runtime_common]",
     "torch",
-    "petit_kernel",
+    "petit_kernel==0.0.2",
 ]
 
 # xpu is not enabled in public vllm and torch whl,