From 9f754180011971589519f50e7b1e2889f6556173 Mon Sep 17 00:00:00 2001 From: baihuawei Date: Fri, 9 Oct 2020 10:24:06 +0800 Subject: [PATCH] fix GPU BroadCast --- .../ccsrc/backend/kernel_compiler/gpu/nccl/nccl_gpu_kernel.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/nccl_gpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/nccl_gpu_kernel.h index 2ffe5fa486..e4aa67b923 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/nccl_gpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/nccl_gpu_kernel.h @@ -112,7 +112,7 @@ class NcclGpuKernel : public GpuKernel { for (int i = 0; i < SizeToInt(input_size_list_.size()); ++i) { input_addr = GetDeviceAddress(inputs, i); output_addr = GetDeviceAddress(outputs, i); - CHECK_NCCL_RET_WITH_EXCEPT((*broadcast_funcptr)(input_addr, output_addr, output_size_ / sizeof(T), + CHECK_NCCL_RET_WITH_EXCEPT((*broadcast_funcptr)(input_addr, output_addr, output_size_list_[i] / sizeof(T), nccl_data_type_, root_, stream, group_name_), "ncclBroadcast failed"); }