p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5740 Reorder gpu fused optimizer 

Merge pull request !5740 from chenweifeng/gpu-momentum-fusion
This commit is contained in:
mindspore-ci-bot 2020-09-04 11:06:21 +08:00 committed by Gitee
parent 4781021c1d 13dd31f56c
commit 2fe99f2bd7
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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,
}; };