提交 305aab03 编写于 作者: D dengwentao

fix embedding look up op run fail

上级 53df6497
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册