From 3f472ce669a32d291d8f87d4924b957cc6256c54 Mon Sep 17 00:00:00 2001 From: wilfChen Date: Tue, 3 Nov 2020 17:41:48 +0800 Subject: [PATCH] fix gpu adam fusion kernel --- .../backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h index 10e90aef8d..9a345682cd 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h @@ -33,7 +33,7 @@ class FusedAdamWeightDecayGpuKernel : public GpuKernel { bool Init(const CNodePtr &kernel_node) override { auto node_name = AnfAlgo::GetCNodeName(kernel_node); - if (node_name == "AdamWeighDecay") { + if (node_name == "FusedAdamWeightDecay") { weight_decay_ = true; }