Browse Source

!1872 fix EmbeddingLookup cpu op run fail

Merge pull request !1872 from dengwentao/fix_cpu_op
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
2ee812d27d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc

+ 1
- 1
mindspore/ccsrc/kernel/cpu/embedding_look_up_cpu_kernel.cc View File

@@ -133,7 +133,7 @@ bool EmbeddingLookUpCPUKernel::Launch(const std::vector<kernel::AddressPtr> &inp
for (int i = 0; i < split_num_; i++) {
device::cpu::MPIAdapter::Instance().ReduceScatter(reinterpret_cast<float *>(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)


Loading…
Cancel
Save