[CI][ROCm] Export Helion benchmark script in test artifacts (#50726)
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Andreas Karatzas committed
Aug 3, 2026 at 13:09 UTC
76d995df2c80fd1f81901723fa05c992714693b2
2 files changed
+3
.buildkite/test-amd.yaml
+1
@@ -163,6 +163,7 @@ steps:
163
working_dir: "/vllm-workspace/tests"
164
source_file_dependencies:
165
- vllm/utils/import_utils.py
166
+ - scripts/benchmark_helion_kernels.py
167
- tests/kernels/helion/
168
- vllm/platforms/rocm.py
169
commands:
docker/Dockerfile.rocm
+2
@@ -325,6 +325,7 @@ COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/requirements /requirements
325
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/benchmarks /benchmarks
326
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/tests /tests
327
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/examples /examples
328
+COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/scripts/benchmark_helion_kernels.py /scripts/benchmark_helion_kernels.py
329
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/tools/install_torchcodec_rocm.sh /tools/install_torchcodec_rocm.sh
330
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/docker/Dockerfile /docker/Dockerfile
331
COPY --from=build_vllm ${COMMON_WORKDIR}/vllm/docker/Dockerfile.cpu /docker/Dockerfile.cpu
@@ -641,6 +642,7 @@ COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/requirements /requir
642
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/benchmarks /benchmarks
643
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/tests /tests
644
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/examples /examples
645
+COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/scripts/benchmark_helion_kernels.py /scripts/benchmark_helion_kernels.py
646
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/tools/install_torchcodec_rocm.sh /tools/install_torchcodec_rocm.sh
647
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/docker/Dockerfile /docker/Dockerfile
648
COPY --from=build_vllm_wheel_release ${COMMON_WORKDIR}/vllm/docker/Dockerfile.cpu /docker/Dockerfile.cpu