Searched defs:rootRank (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ops_nccl.py | 103 def broadcast(xs, rootRank, rootTensor): argument
323 def reduce(xs, rootRank, rootTensor, op=None): argument
474 def gather(output_t, input_t, rootRank): argument
509 def gather(output_t, input_t, rootRank): argument
587 def scatter(output_t, input_t, rootRank): argument
622 def scatter(output_t, input_t, rootRank): argument
|
| H A D | test_c10d_ucc.py | 177 def broadcast(xs, rootRank, rootTensor): argument
|
| H A D | test_c10d_gloo.py | 319 def broadcast(xs, rootRank, rootTensor): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Types.hpp | 115 int64_t rootRank = 0; member
131 int64_t rootRank = 0; member
142 int64_t rootRank = 0; member
147 int64_t rootRank = 0; member
|
| H A D | init.cpp | 1834 int rootRank) { in c10d_init()
1894 const ::c10d::ReduceOp& op) { in c10d_init()
1959 int rootRank) { in c10d_init()
1982 int rootRank) { in c10d_init()
2279 int rootRank) { in c10d_init()
2336 const ::c10d::ReduceOp& op) { in c10d_init()
2393 int rootRank) { in c10d_init()
2416 int rootRank) { in c10d_init()
|
| H A D | ProcessGroupGloo.cpp | 874 int rootRank, in AsyncBroadcastWork()
892 const int rootRank; member in c10d::__anon2d5ad4bf0711::AsyncBroadcastWork
923 int rootRank, in AsyncBroadcastCUDAWork()
1624 int rootRank, in AsyncReduceWork()
1644 const int rootRank; member in c10d::__anon2d5ad4bf0e11::AsyncReduceWork
1683 int rootRank, in AsyncReduceCUDAWork()
|
| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupNCCLTest.cpp | 270 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
289 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
536 for (const auto rootRank : c10::irange(size)) { in testBroadcast() local
563 for (const auto rootRank : c10::irange(size)) { in testReduce() local
697 for (const auto rootRank : c10::irange(size)) { in testSplittingCommunicator() local
|
| H A D | ProcessGroupGlooAsyncTest.cpp | 158 c10::intrusive_ptr<c10d::Work> run(int rootRank, int rootTensor) { in run()
229 for (const auto rootRank : c10::irange(numProcesses)) { in runAsyncBroadcastTest() local
|