.. |
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
8 months ago |
cuda_wrapper.py
|
9c0e7d95c8
fix: libcudart path for some versions of pytorch (#726)
|
3 months ago |
custom_all_reduce.py
|
239a8cae25
torch.compile: register all-reduce operations as custom ops (#1050)
|
1 week ago |
custom_all_reduce_utils.py
|
2940da2c7b
distributed: fix custom allreduce p2p cache file generation (#922)
|
3 weeks ago |
pynccl.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 months ago |
pynccl_wrapper.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 months ago |
shm_broadcast.py
|
f81e7d7010
distributed: bind only to 127.0.0.1 for local-only usage (#1061)
|
1 week ago |
tpu_communicator.py
|
61103b92d4
tpu: support single and multi-host TPUs on GKE and RayServe (#970)
|
2 weeks ago |