mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!38268 解决ApplyAdamWithAmsgrad算子的r1.8构建告警问题-同步更新master 

Merge pull request !38268 from OwenSec/master
This commit is contained in:
i-robot 2022-07-18 08:09:52 +00:00 committed by Gitee
parent 0a49f09637 3e427d6089
commit 058f6a2577
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.cc
View File

@ -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"

2
mindspore/ccsrc/plugin/device/cpu/kernel/apply_adam_with_amsgrad_cpu_kernel.h
View File

@ -35,7 +35,7 @@ class ApplyAdamWithAmsgradCpuKernelMod : public NativeCpuKernelMod {
const std::vector<KernelTensorPtr> &outputs) override;
int Resize(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
const std::vector<KernelTensorPtr> &outputs, const std::map<uint32_t, tensor::TensorPtr> &) override;
const std::vector<KernelTensorPtr> &outputs, const std::map<uint32_t, tensor::TensorPtr> &others) override;
bool Launch(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &workspace,
const std::vector<AddressPtr> &outputs) override;