diff --git a/.buildkite/test-amd.yaml b/.buildkite/test-amd.yaml index b7254efd2dc2..f42c495f864c 100644 --- a/.buildkite/test-amd.yaml +++ b/.buildkite/test-amd.yaml @@ -790,7 +790,7 @@ steps: - tests/kernels/helion/ - vllm/platforms/rocm.py commands: - - pip install helion + - pip install helion==0.3.3 - pytest -v -s kernels/helion/ diff --git a/.buildkite/test_areas/kernels.yaml b/.buildkite/test_areas/kernels.yaml index 69abc69b0fbc..aecb40f1838b 100644 --- a/.buildkite/test_areas/kernels.yaml +++ b/.buildkite/test_areas/kernels.yaml @@ -139,7 +139,7 @@ steps: - vllm/utils/import_utils.py - tests/kernels/helion/ commands: - - pip install helion + - pip install helion==0.3.3 - pytest -v -s kernels/helion/ diff --git a/setup.py b/setup.py index 74997702950e..0d478c59a231 100644 --- a/setup.py +++ b/setup.py @@ -1063,7 +1063,10 @@ def _read_requirements(filename: str) -> list[str]: # Optional deps for AMD FP4 quantization support "petit-kernel": ["petit-kernel"], # Optional deps for Helion kernel development - "helion": ["helion==0.3.2"], + # NOTE: When updating helion version, also update CI files: + # - .buildkite/test_areas/kernels.yaml + # - .buildkite/test-amd.yaml + "helion": ["helion==0.3.3"], # Optional deps for gRPC server (vllm serve --grpc) "grpc": ["smg-grpc-servicer[vllm] >= 0.5.0"], # Optional deps for OpenTelemetry tracing