Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Commit

Permalink
[ci][test] fix ca test in main (vllm-project#5746)
Browse files Browse the repository at this point in the history
  • Loading branch information
youkaichao authored and robertgshaw2-redhat committed Jun 23, 2024
1 parent 07cd29d commit 2e2140f
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions .buildkite/test-pipeline.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -197,6 +197,9 @@ steps:
gpu: a100
num_gpus: 4
commands:
# FIXIT: find out which code initialize cuda before running the test
# before the fix, we need to use spawn to test it
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
# NOTE: don't test llama model here, it seems hf implementation is buggy
# see https://github.com/vllm-project/vllm/pull/5689 for details
- pytest -v -s distributed/test_custom_all_reduce.py
Expand Down

0 comments on commit 2e2140f

Please sign in to comment.