Browse Source

!8173 fix gpu adam kernel fusion

Merge pull request !8173 from chenweifeng/fix-gpu-adam-fusion
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cf64fe4e07
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/fused_adam_weight_decay.h View File

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



Loading…
Cancel
Save