Home
last modified time | relevance | path

Searched refs:nccl_dtype (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h52 ncclDataType_t nccl_dtype(const TypeId &type_id) { return kNcclDtypeMap[TypeIdLabel(type_id)]; } in nccl_dtype() function
Dsync_batch_norm_grad_gpu_kernel.h72 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()
180 … (*all_reduce_funcptr)(input_addr, output_addr, C_, nccl_dtype(kNumberTypeFloat32), in LaunchAllReduce()
Dsync_batch_norm_gpu_kernel.h85 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()
215 …(*all_gather_funcptr)(input_addr, output_addr, C_, nccl_dtype(GetTypeID(input_addr)), stream, grou… in LaunchAllGather()
Dnccl_send_gpu_kernel.h63 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()
Dnccl_recv_gpu_kernel.h67 nccl_data_type_ = nccl_dtype(AnfAlgo::GetOutputDeviceDataType(kernel_node, 0)); in Init()
Dnccl_collective_gpu_kernel.h85 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()