From 3e427d6089d9db579bd42e110d8d27f7698eed5d Mon Sep 17 00:00:00 2001 From: OwenSec Date: Mon, 18 Jul 2022 09:39:23 +0800 Subject: [PATCH] amsgrad cpu kernel correct Resize definition delete errorcode.h --- .../device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc | 1 - .../device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc index 913ab22e6d2..3ad3782a1c7 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc @@ -21,7 +21,6 @@ #include "kernel/common_utils.h" #include "plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h" #include "plugin/device/cpu/hal/device/cpu_device_address.h" -#include "plugin/device/cpu/kernel/nnacl/errorcode.h" #include "plugin/device/cpu/kernel/nnacl/fp32/adam_fp32.h" #include "mindspore/core/ops/apply_adam_with_amsgrad.h" #include "utils/ms_utils.h" diff --git a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h index 96d17090d4b..b348e12e0e7 100644 --- a/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h +++ b/mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h @@ -35,7 +35,7 @@ class ApplyAdamWithAmsgradCpuKernelMod : public NativeCpuKernelMod { const std::vector &outputs) override; int Resize(const BaseOperatorPtr &base_operator, const std::vector &inputs, - const std::vector &outputs, const std::map &) override; + const std::vector &outputs, const std::map &others) override; bool Launch(const std::vector &inputs, const std::vector &workspace, const std::vector &outputs) override;