Searched refs:LaunchAllReduce (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | sync_batch_norm_grad_gpu_kernel.h | 60 LaunchAllReduce(dy_sum_local, dy_sum_red, stream_ptr); in Launch()
61 LaunchAllReduce(dot_p_local, dot_p_red, stream_ptr); in Launch()
175 void LaunchAllReduce(reduce_type *input_addr, reduce_type *output_addr, void *stream_ptr) { in LaunchAllReduce() function
|
| D | nccl_collective_gpu_kernel.h | 60 LaunchAllReduce(inputs, outputs, stream_ptr); in Launch()
156 …void LaunchAllReduce(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &outputs, in LaunchAllReduce() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_bucket.h | 37 void LaunchAllReduce() override;
|
| D | gpu_bucket.cc | 121 void GPUBucket::LaunchAllReduce() { in LaunchAllReduce() function in mindspore::device::gpu::GPUBucket
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_bucket.h | 37 void LaunchAllReduce() override;
|
| D | ascend_bucket.cc | 123 void AscendBucket::LaunchAllReduce() { in LaunchAllReduce() function in mindspore::device::ascend::AscendBucket
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | bucket.h | 83 virtual void LaunchAllReduce() = 0;
|
| D | bucket.cc | 53 LaunchAllReduce(); in Launch()
|