Browse Source

!5740 Reorder gpu fused optimizer

Merge pull request !5740 from chenweifeng/gpu-momentum-fusion
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
2fe99f2bd7
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/ccsrc/utils/utils.h

+ 4
- 0
mindspore/ccsrc/utils/utils.h View File

@@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
kApplyProximalAdagradOpName, kApplyProximalAdagradOpName,
kApplyProximalGradientDescentOpName, kApplyProximalGradientDescentOpName,
kApplyRMSPropOpName, kApplyRMSPropOpName,
kFusedAdamWeightDecayName,
kFusedAdamName,
kFusedWeightScaleApplyMomentum,
kFusedScaleApplyMomentum,
kPullOpName, kPullOpName,
}; };




Loading…
Cancel
Save