Searched refs:AsyncSparseAllreduceCUDAWork (Results 1 – 1 of 1) sorted by relevance
1430 class AsyncSparseAllreduceCUDAWork : public AsyncSparseAllreduceWork { class1432 AsyncSparseAllreduceCUDAWork( in AsyncSparseAllreduceCUDAWork() function in c10d::__anon725d3f3f0911::AsyncSparseAllreduceCUDAWork1535 work = c10::make_intrusive<AsyncSparseAllreduceCUDAWork>( in allreduce()