From 9a0ae7dddeac70d93f9b1dac2d18b292d3a1bae0 Mon Sep 17 00:00:00 2001
From: alouhahaha <wangjun289@huawei.com>
Date: Tue, 9 Mar 2021 14:15:45 +0800
Subject: [PATCH] modify the data format of split_dims

---
 .../ascend/enhancer/split_inputs_for_reduce_scatter.cc          | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc b/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc
index de34a05f1f3..44d5e7eff71 100644
--- a/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc
+++ b/mindspore/ccsrc/backend/optimizer/ascend/enhancer/split_inputs_for_reduce_scatter.cc
@@ -42,7 +42,7 @@ std::vector<AnfNodePtr> SplitInputsForReduceScatter::InsertSplitForInput(const F
     }
     AnfAlgo::SetOutputInferTypeAndShape(dtypes, shapes, split.get());
 
-    AnfAlgo::SetNodeAttr("split_dim", MakeValue(0), split);
+    AnfAlgo::SetNodeAttr("split_dim", MakeValue(0L), split);
     AnfAlgo::SetNodeAttr("num_split", MakeValue(SizeToInt(rank_size)), split);
     AnfAlgo::SetNodeAttr("size_splits", MakeValue(size_splits), split);
     kernel_select_->SelectKernel(split);