Files
vllm/tests/distributed/test_pipeline_partition.py