From a6078902566cc73b310288202c0c07014e412304 Mon Sep 17 00:00:00 2001
From: jjfeing <jiangjianfei3@huawei.com>
Date: Mon, 23 Nov 2020 20:25:43 +0800
Subject: [PATCH] fix_return_scalar

---
 .../backend/optimizer/pass/convert_const_scalar_to_tensor.cc  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc b/mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc
index 7ba67ff74f..f9a93d25a1 100644
--- a/mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc
+++ b/mindspore/ccsrc/backend/optimizer/pass/convert_const_scalar_to_tensor.cc
@@ -60,6 +60,10 @@ const AnfNodePtr ConvertConstScalarToTensor::Process(const FuncGraphPtr &func_gr
   if (node == nullptr || func_graph == nullptr || AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem)) {
     return nullptr;
   }
+  // input is scalar, and link to graph return
+  if (node->isa<ValueNode>() && node == func_graph->output()) {
+    return CreateTensorInput(func_graph->cast<KernelGraphPtr>(), node);
+  }
   if (!node->isa<CNode>()) {
     return nullptr;
   }