Searched refs:all_reduce_funcptr (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
D | gpu_bucket.cc | 124 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/ |
D | sync_batch_norm_grad_gpu_kernel.h | 177 …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()
|
D | nccl_collective_gpu_kernel.h | 161 …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()
|