Murali Andoorveedu 0f6d7a9a34 [Models] Add remaining model PP support (#7168) ... Signed-off-by: Muralidhar Andoorveedu <muralidhar.andoorveedu@centml.ai> Signed-off-by: Murali Andoorveedu <muralidhar.andoorveedu@centml.ai> Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>		2024-10-04 10:56:58 +08:00
..
__init__.py	[CI/Build] Move test_utils.py to tests/utils.py (#4425)	2024-05-13 23:50:09 +09: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_multi_node_assignment.py	[Ray backend] Better error when pg topology is bad. (#7584)	2024-08-22 17:44:25 -07:00
test_pipeline_parallel.py	[Models] Add remaining model PP support (#7168)	2024-10-04 10:56:58 +08: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/Build] Reorganize models tests (#7820)	2024-09-13 10:20:06 -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