From 74c7bdd4716fd743ccad1a4225cb0e4ca2800799 Mon Sep 17 00:00:00 2001
From: kswang <wangkaisheng2@huawei.com>
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<std::string> kOptOperatorSet = {
   kFusedSparseAdamName,
   kFusedWeightScaleApplyMomentum,
   kFusedScaleApplyMomentum,
+  kApplyCenteredRMSPropOpName,
+  kFusedSparseFtrlName,
+  kFusedSparseProximalAdagradName,
+  kFusedSparseLazyAdamName,
+  kSparseApplyFtrlName,
+  kSparseApplyFtrlV2Name,
+  kSGDName,
+  kLARSUpdateName,
   kPullOpName,
 };