Home
last modified time | relevance | path

Searched defs:LaunchAllReduce (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc121 void GPUBucket::LaunchAllReduce() { in LaunchAllReduce() function in mindspore::device::gpu::GPUBucket
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.cc123 void AscendBucket::LaunchAllReduce() { in LaunchAllReduce() function in mindspore::device::ascend::AscendBucket
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dsync_batch_norm_grad_gpu_kernel.h175 void LaunchAllReduce(reduce_type *input_addr, reduce_type *output_addr, void *stream_ptr) { in LaunchAllReduce() function
Dnccl_collective_gpu_kernel.h156 …void LaunchAllReduce(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &outputs, in LaunchAllReduce() function