From 74c7bdd4716fd743ccad1a4225cb0e4ca2800799 Mon Sep 17 00:00:00 2001 From: kswang Date: Sat, 24 Oct 2020 14:35:02 +0800 Subject: [PATCH] fix segmentfault with fused sparse ftrl --- mindspore/ccsrc/utils/utils.h | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/mindspore/ccsrc/utils/utils.h b/mindspore/ccsrc/utils/utils.h index 26f39625a2..02a3279e32 100644 --- a/mindspore/ccsrc/utils/utils.h +++ b/mindspore/ccsrc/utils/utils.h @@ -223,6 +223,13 @@ constexpr auto kBasicLSTMCellInputGradOpName = "BasicLSTMCellInputGrad"; constexpr auto kBasicLSTMCellOpName = "BasicLSTMCell"; constexpr auto kDynamicRNNOpName = "DynamicRNN"; constexpr auto kLSTMInputGradOpName = "LSTMInputGrad"; +constexpr auto kFusedSparseFtrlName = "FusedSparseFtrl"; +constexpr auto kFusedSparseProximalAdagradName = "FusedSparseProximalAdagrad"; +constexpr auto kFusedSparseLazyAdamName = "FusedSparseLazyAdam"; +constexpr auto kSparseApplyFtrlName = "SparseApplyFtrl"; +constexpr auto kSparseApplyFtrlV2Name = "SparseApplyFtrlV2"; +constexpr auto kSGDName = "SGD"; +constexpr auto kLARSUpdateName = "LARSUpdate"; // Hcom Op Type constexpr auto kHcomOpTypeAllReduce = "HcomAllReduce"; @@ -389,6 +396,14 @@ const std::set kOptOperatorSet = { kFusedSparseAdamName, kFusedWeightScaleApplyMomentum, kFusedScaleApplyMomentum, + kApplyCenteredRMSPropOpName, + kFusedSparseFtrlName, + kFusedSparseProximalAdagradName, + kFusedSparseLazyAdamName, + kSparseApplyFtrlName, + kSparseApplyFtrlV2Name, + kSGDName, + kLARSUpdateName, kPullOpName, };