From 305aab03f74b87f739682e1ad6630782b57a952b Mon Sep 17 00:00:00 2001 From: dengwentao Date: Sat, 6 Jun 2020 10:55:31 +0800 Subject: [PATCH] fix embedding look up op run fail --- mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc b/mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc index d86e49fc58..e56afaf9b5 100644 --- a/mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc +++ b/mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc @@ -133,7 +133,7 @@ bool EmbeddingLookUpCPUKernel::Launch(const std::vector &inp for (int i = 0; i < split_num_; i++) { device::cpu::MPIAdapter::Instance().ReduceScatter(reinterpret_cast(gather_v2_out_) + i * one_split_lens, output_addr + i * reduce_scatter_out_lens, group, - one_split_lens, "sum"); + one_split_lens / 8, "sum"); } } #if defined(_WIN32) || defined(_WIN64)