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); }