Peter Salas 1ca0d4f86b [Model] Add UltravoxModel and UltravoxConfig (#7615)

2024-08-21 22:49:39 +00: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

[Model] Add UltravoxModel and UltravoxConfig (#7615)

2024-08-21 22:49:39 +00:00

test_basic_distributed_correctness.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -07:00

test_chunked_prefill_distributed.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -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