[aDAG] Unflake aDAG + PP tests (#7600)

2024-08-16 20:49:30 -07:00
parent bae888cb8e
commit 4706eb628e
3 changed files with 9 additions and 4 deletions
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -314,11 +314,11 @@ steps:
  num_gpus: 4
  source_file_dependencies:
  - vllm/
-  - tests/distributed/test_pipeline_parallel
  - tests/distributed/test_pp_cudagraph.py
+  - tests/distributed/test_pipeline_parallel
  commands:
-  - pytest -v -s distributed/test_pipeline_parallel.py
  - pytest -v -s distributed/test_pp_cudagraph.py
+  - pytest -v -s distributed/test_pipeline_parallel.py

 - label: LoRA Long Context (Distributed) # 11min
  # This test runs llama 13B, so it is required to run on 4 GPUs.