From 3e8f3470a31e18d00d694fdede120721dc7158b8 Mon Sep 17 00:00:00 2001
From: zhangxiaokun <zhang.xiaokun@huawei.com>
Date: Fri, 30 Apr 2021 11:27:34 +0800
Subject: [PATCH] SetTaskContext for known

---
 ge/hybrid/executor/subgraph_executor.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ge/hybrid/executor/subgraph_executor.cc b/ge/hybrid/executor/subgraph_executor.cc
index c9c6f768..60895c7e 100644
--- a/ge/hybrid/executor/subgraph_executor.cc
+++ b/ge/hybrid/executor/subgraph_executor.cc
@@ -177,6 +177,7 @@ Status SubgraphExecutor::ExecuteAsyncForKnownShape(const std::vector<TensorValue
 
   known_shape_task_context_ = TaskContext::Create(node_state.get(), context_, subgraph_context_.get());
   GE_CHECK_NOTNULL(known_shape_task_context_);
+  node_state->SetTaskContext(known_shape_task_context_);
 
   std::function<void()> callback;
   GE_CHK_STATUS_RET_NOLOG(InitCallback(node_state.get(), callback));