vllm/tests/distributed
2024-08-16 20:49:30 -07:00
..
__init__.py [CI/Build] Move test_utils.py to tests/utils.py (#4425) 2024-05-13 23:50:09 +09:00
test_basic_distributed_correctness_enc_dec.py [Core] Support serving encoder/decoder models (#7258) 2024-08-09 10:39:41 +08:00
test_basic_distributed_correctness.py [ci][distributed] merge distributed test commands (#7097) 2024-08-02 21:33:53 -07:00
test_chunked_prefill_distributed.py [ci][distributed] merge distributed test commands (#7097) 2024-08-02 21:33:53 -07:00
test_comm_ops.py [Misc/Testing] Use torch.testing.assert_close (#7324) 2024-08-16 04:24:04 +00:00
test_custom_all_reduce.py [Misc/Testing] Use torch.testing.assert_close (#7324) 2024-08-16 04:24:04 +00:00
test_distributed_oot.py [misc][plugin] add plugin system implementation (#7426) 2024-08-13 16:24:17 -07:00
test_multimodal_broadcast.py [Bugfix] Fix weight loading for Chameleon when TP>1 (#7410) 2024-08-13 05:33:41 +00:00
test_pipeline_parallel.py [aDAG] Unflake aDAG + PP tests (#7600) 2024-08-16 20:49:30 -07:00
test_pipeline_partition.py [MISC] Introduce pipeline parallelism partition strategies (#6920) 2024-07-31 12:02:17 -07:00
test_pp_cudagraph.py [ci/test] rearrange tests and make adag test soft fail (#7572) 2024-08-15 19:39:04 -07:00
test_pynccl.py [Distributed] Add send and recv helpers (#5719) 2024-06-23 14:42:28 -07:00
test_same_node.py [ci] try to add multi-node tests (#6280) 2024-07-12 21:51:48 -07:00
test_shm_broadcast.py [core][distributed] zmq fallback for broadcasting large objects (#6183) 2024-07-09 18:49:11 -07:00
test_utils.py [ROCm][AMD] unify CUDA_VISIBLE_DEVICES usage in cuda/rocm (#6352) 2024-07-11 21:30:46 -07:00