Skip to content

Commit d5b73f8

Browse files
committed
add ci workflow
1 parent 68aaaaf commit d5b73f8

File tree

1 file changed

+24
-1
lines changed

1 file changed

+24
-1
lines changed

.github/workflows/pre-merge.yaml

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,9 @@ jobs:
4747
RUN git checkout main
4848
4949
# Pinning versions in requirements might be good practice for CI consistency
50-
RUN pip install pytest pytest_asyncio
50+
RUN pip install pytest pytest_asyncio nixl==0.4.1
5151
RUN pip install git+https://github.com/EleutherAI/lm-evaluation-harness.git
52+
RUN pip install lm-eval[api]
5253
5354
ENV no_proxy=localhost,127.0.0.1
5455
ENV PT_HPU_ENABLE_LAZY_COLLECTIVES=true
@@ -94,7 +95,29 @@ jobs:
9495
9596
EXITCODE=$?
9697
echo "Test script exited with code: $EXITCODE"
98+
hpu_pd_tests:
99+
runs-on: ucb-vllm-cicd-g2
100+
# This is a final job that runs after the build and unit tests
101+
# Unit tests are supposed to fail fast if anything goes wrong, removing the need for this job to run.
102+
needs: [pre_merge_hpu_test_build, hpu_unit_tests]
103+
steps:
104+
- name: Run test scripts
105+
run: |
106+
EXITCODE=1
107+
remove_docker_containers() { docker rm -f hpu-plugin-v1-test-pd-tests-${{ github.event.pull_request.head.sha }} || true; }
108+
trap 'remove_docker_containers; exit $EXITCODE;' EXIT
109+
remove_docker_containers
110+
111+
echo "Running HPU plugin v1 nixl pd tests"
112+
docker run --rm --runtime=habana --name=hpu-plugin-v1-test-pd-tests-${{ github.event.pull_request.head.sha }} --network=host \
113+
-e HABANA_VISIBLE_DEVICES=all \
114+
-e HF_HOME=/workspace/hf_cache \
115+
-v /mnt/hf_cache:/workspace/hf_cache \
116+
hpu-plugin-v1-test-env-pre-merge-${{ github.event.pull_request.head.sha }} \
117+
/bin/bash "/workspace/vllm-gaudi/tests/v1/kv_connector/nixl_integration/run_accuracy_test.sh"
97118
119+
EXITCODE=$?
120+
echo "Test script exited with code: $EXITCODE"
98121
hpu_e2e_tests:
99122
runs-on: ucb-vllm-cicd-g2
100123
# This is a final job that runs after the build and unit tests

0 commit comments

Comments
 (0)