diff --git a/.buildkite/test-amd.yaml b/.buildkite/test-amd.yaml index b7254efd2..f42c495f8 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 69abc69b0..aecb40f18 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 749977029..0d478c59a 100644 --- a/setup.py +++ b/setup.py @@ -1063,7 +1063,10 @@ setup( # 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