!5740 Reorder gpu fused optimizer

Merge pull request !5740 from chenweifeng/gpu-momentum-fusion
2020-09-04 11:06:21 +08:00 · 2020-09-04 11:06:21 +08:00 · 2fe99f2bd7
parent 4781021c1d 13dd31f56c
commit 2fe99f2bd7
1 changed files with 4 additions and 0 deletions
--- a/mindspore/ccsrc/utils/utils.h
+++ b/mindspore/ccsrc/utils/utils.h
@ -344,6 +344,10 @@ const std::set<std::string> kOptOperatorSet = {
  kApplyProximalAdagradOpName,
  kApplyProximalGradientDescentOpName,
  kApplyRMSPropOpName,
+  kFusedAdamWeightDecayName,
+  kFusedAdamName,
+  kFusedWeightScaleApplyMomentum,
+  kFusedScaleApplyMomentum,
  kPullOpName,
 };