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