Shane A 9db713a1dc

[Model] Add OLMo November 2024 model (#10503)

2024-11-25 17:26:40 -05:00

..

__init__.py

…

test_ca_buffer_sharing.py

[distributed] add function to create ipc buffers directly (#10064)

2024-11-05 22:35:03 -08: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

[Core][Distributed] Refactor ipc buffer init in CustomAllreduce (#10030)

2024-11-06 23:50:47 -08: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

[Model] Add OLMo November 2024 model (#10503)

2024-11-25 17:26:40 -05: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

[Misc] Add pynccl wrappers for all_gather and reduce_scatter (#9432)

2024-11-22 22:14:03 -05:00

test_same_node.py

[CI/Build] Reorganize models tests (#7820)

2024-09-13 10:20:06 -07:00

test_shm_broadcast.py

…

test_utils.py

[ci][distributed] disable hanging tests (#10317)

2024-11-14 00:23:39 -08:00