From 8ba99874605adadec2408d01b7aed4acf1ec58d1 Mon Sep 17 00:00:00 2001 From: yang_chun Date: Thu, 14 Jan 2021 17:35:03 +0800 Subject: [PATCH] Fix: ApplyAdagrad CodeDEX. --- .../backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.cc | 4 +++- .../backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.h | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.cc index 90d7c9d482..34d4c1ee16 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.cc @@ -84,6 +84,7 @@ void ApplyAdagradCPUKernel::LaunchKernel(const std::vector &inputs) if (batch_size == 0) { MS_LOG(EXCEPTION) << "Error occur in launch kernel"; + return; } while (start < length) { size_t end = (start + batch_size) > length ? length : (start + batch_size); @@ -98,7 +99,8 @@ void ApplyAdagradCPUKernel::LaunchKernel(const std::vector &inputs) } template -void ApplyAdagradCPUKernel::LaunchApplyAdagrad(T var, T accum, T lr, T gradient, size_t start, size_t end) { +void ApplyAdagradCPUKernel::LaunchApplyAdagrad(T const var, T const accum, const T lr, const T gradient, size_t start, + size_t end) { // DataType can only be float32 or float16, so eps will not be zero. using DataType = typename std::iterator_traits::value_type; const DataType one = DataType(1); diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.h index 739c964c10..f5ebe37fbf 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/apply_adagrad_cpu_kernel.h @@ -38,7 +38,7 @@ class ApplyAdagradCPUKernel : public CPUKernel { template void LaunchKernel(const std::vector &inputs); template - void LaunchApplyAdagrad(T var, T accum, T lr, T gradient, size_t start, size_t end); + void LaunchApplyAdagrad(T const var, T const accum, const T lr, const T gradient, size_t start, size_t end); bool update_slots_{true}; TypeId dtype_{kTypeUnknown}; };