Home
last modified time | relevance | path

Searched refs:AsyncSparseAllreduceWork (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupGloo.cpp1102 class AsyncSparseAllreduceWork : public ProcessGroupGloo::AsyncWork { class
1104 AsyncSparseAllreduceWork( in AsyncSparseAllreduceWork() function in c10d::__anon2d5ad4bf0911::AsyncSparseAllreduceWork
1430 class AsyncSparseAllreduceCUDAWork : public AsyncSparseAllreduceWork {
1437 : AsyncSparseAllreduceWork(context, inputs, tag, seq) { in AsyncSparseAllreduceCUDAWork()
1525 work = c10::make_intrusive<AsyncSparseAllreduceWork>( in allreduce()