Searched refs:nccl_dtype (Results 1 – 6 of 6) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
D | nccl_gpu_kernel.h | 52 ncclDataType_t nccl_dtype(const TypeId &type_id) { return kNcclDtypeMap[TypeIdLabel(type_id)]; } in nccl_dtype() function
|
D | sync_batch_norm_grad_gpu_kernel.h | 72 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()
|
D | sync_batch_norm_gpu_kernel.h | 85 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()
|
D | nccl_send_gpu_kernel.h | 63 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()
|
D | nccl_recv_gpu_kernel.h | 67 nccl_data_type_ = nccl_dtype(AnfAlgo::GetOutputDeviceDataType(kernel_node, 0)); in Init()
|
D | nccl_collective_gpu_kernel.h | 85 nccl_data_type_ = nccl_dtype(AnfAlgo::GetInputDeviceDataType(kernel_node, 0)); in Init()
|