From 90f06ffde578a4c15cebc07868b67ca27ceda279 Mon Sep 17 00:00:00 2001 From: ZPaC Date: Mon, 14 Dec 2020 09:33:13 +0800 Subject: [PATCH] Fix bug while there are multiple embedding-lookup --- .../kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc | 3 +++ .../kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc | 3 +++ .../kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc | 3 +++ 3 files changed, 9 insertions(+) diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc index 084e44aed2..b24ce98f8c 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_adam_ps_kernel.cc @@ -82,6 +82,9 @@ void SparseApplyAdamPSKernel::ReInit(const std::vector &inputs) { bool SparseApplyAdamPSKernel::Execute(const std::vector &inputs, const std::vector &workspace, const std::vector &outputs) { ReInit(inputs); + if (indices_size_ == 0) { + return true; + } return Launch(inputs, workspace, outputs); } diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc index 8949745065..c1ee4b143a 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_ftrl_ps_kernel.cc @@ -89,6 +89,9 @@ void SparseApplyFtrlPSKernel::ReInit(const std::vector &inputs) { bool SparseApplyFtrlPSKernel::Execute(const std::vector &inputs, const std::vector &workspace, const std::vector &outputs) { ReInit(inputs); + if (indices_size_ == 0) { + return true; + } return Launch(inputs, workspace, outputs); } diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc index b43c132ba4..a296330c27 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/sparse_apply_lazy_adam_ps_kernel.cc @@ -82,6 +82,9 @@ bool SparseApplyLazyAdamPSKernel::Execute(const std::vector &inputs, const std::vector &workspace, const std::vector &outputs) { ReInit(inputs); + if (indices_size_ == 0) { + return true; + } return Launch(inputs, workspace, outputs); }