Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc124 auto all_reduce_funcptr = in LaunchAllReduce() local
126 MS_EXCEPTION_IF_NULL(all_reduce_funcptr); in LaunchAllReduce()
150 …(*all_reduce_funcptr)(ar_input_addr_, ar_output_addr_, total_size_ / type_size, nccl_data_type_ite… in LaunchAllReduce()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dsync_batch_norm_grad_gpu_kernel.h177 …auto all_reduce_funcptr = reinterpret_cast<AllReduce>(dlsym(const_cast<void *>(collective_handle_)… in LaunchAllReduce() local
178 MS_EXCEPTION_IF_NULL(all_reduce_funcptr); in LaunchAllReduce()
180 … (*all_reduce_funcptr)(input_addr, output_addr, C_, nccl_dtype(kNumberTypeFloat32), in LaunchAllReduce()
Dnccl_collective_gpu_kernel.h161 …auto all_reduce_funcptr = reinterpret_cast<AllReduce>(dlsym(const_cast<void *>(collective_handle_)… in LaunchAllReduce() local
162 MS_EXCEPTION_IF_NULL(all_reduce_funcptr); in LaunchAllReduce()
164 … (*all_reduce_funcptr)(input_addr, output_addr, output_size_ / sizeof(T), nccl_data_type_, in LaunchAllReduce()