Murali Andoorveedu c5832d2ae9 [Core] Pipeline Parallel Support (#4412) ... Signed-off-by: Muralidhar Andoorveedu <muralidhar.andoorveedu@centml.ai>		2024-07-02 10:58:08 -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.py	[ci][distributed] fix device count call	2024-06-30 08:06:13 +00:00
test_chunked_prefill_distributed.py	[ci][distributed] fix device count call	2024-06-30 08:06:13 +00:00
test_comm_ops.py	[Core] Pipeline Parallel Support (#4412)	2024-07-02 10:58:08 -07:00
test_custom_all_reduce.py	[Distributed] Add send and recv helpers (#5719)	2024-06-23 14:42:28 -07:00
test_multimodal_broadcast.py	[CI/Build] Add TP test for vision models (#5892)	2024-06-29 15:45:54 +00:00
test_parallel_state.py	[Distributed] Make it clear that % should not be in tensor dict keys. (#5927)	2024-06-28 15:20:22 +00:00
test_pipeline_parallel.py	[Core] Pipeline Parallel Support (#4412)	2024-07-02 10:58:08 -07:00
test_pynccl.py	[Distributed] Add send and recv helpers (#5719)	2024-06-23 14:42:28 -07:00
test_same_node.py	[Core][Distributed] add same-node detection (#5369)	2024-06-11 10:53:59 -07:00
test_shm_broadcast.py	[bugfix][distributed] fix shm broadcast when the queue size is full (#5801)	2024-06-25 21:56:02 -07:00
test_utils.py	[Hardware][AMD][CI/Build][Doc] Upgrade to ROCm 6.1, Dockerfile improvements, test fixes (#5422)	2024-06-25 15:56:15 -07:00