From dd227923440e8b9ad4a289a6891d074fc292fd76 Mon Sep 17 00:00:00 2001
From: zhoufeng <zhoufeng54@huawei.com>
Date: Tue, 23 Jun 2020 19:14:39 +0800
Subject: [PATCH] Graph kernel use control sink

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
---
 mindspore/ccsrc/session/session_basic.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/session/session_basic.cc b/mindspore/ccsrc/session/session_basic.cc
index 47d726451e..12a7b397c7 100644
--- a/mindspore/ccsrc/session/session_basic.cc
+++ b/mindspore/ccsrc/session/session_basic.cc
@@ -429,7 +429,12 @@ CNodePtr SessionBasic::CreateNewCNode(const CNodePtr &cnode, KernelGraph *graph)
   std::vector<AnfNodePtr> cnode_inputs;
   auto attr_input = cnode->input(kAnfPrimitiveIndex);
   MS_EXCEPTION_IF_NULL(attr_input);
-  if (IsValueNode<FuncGraph>(attr_input)) {
+  if (AnfAlgo::IsGraphKernel(cnode)) {
+    auto fg = AnfAlgo::GetCNodeFuncGraphPtr(cnode);
+    MS_EXCEPTION_IF_NULL(fg);
+    auto new_fg = BasicClone(fg);
+    cnode_inputs.push_back(std::make_shared<ValueNode>(new_fg));
+  } else if (IsValueNode<FuncGraph>(attr_input)) {
     // create primitive of cnode:call
     cnode_inputs = {graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name())))};
     // create a ValueNode<KernelGraph> as input of cnode:call