Browse Source

!9884 Fix bug while there are multiple embedding-lookup

From: @zpac
Reviewed-by: @limingqi107,@kisnwang
Signed-off-by: @kisnwang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
6a5cf1c711
3 changed files with 9 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc
  2. +3
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc
  3. +3
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc

+ 3
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc View File

@@ -82,6 +82,9 @@ void SparseApplyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) {
bool SparseApplyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
const std::vector<AddressPtr> &outputs) {
ReInit(inputs);
if (indices_size_ == 0) {
return true;
}
return Launch(inputs, workspace, outputs);
}



+ 3
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc View File

@@ -89,6 +89,9 @@ void SparseApplyFtrlPSKernel::ReInit(const std::vector<AddressPtr> &inputs) {
bool SparseApplyFtrlPSKernel::Execute(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
const std::vector<AddressPtr> &outputs) {
ReInit(inputs);
if (indices_size_ == 0) {
return true;
}
return Launch(inputs, workspace, outputs);
}



+ 3
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc View File

@@ -82,6 +82,9 @@ bool SparseApplyLazyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs,
const std::vector<AddressPtr> &workspace,
const std::vector<AddressPtr> &outputs) {
ReInit(inputs);
if (indices_size_ == 0) {
return true;
}
return Launch(inputs, workspace, outputs);
}



Loading…
Cancel
Save