diff --git a/ge/hybrid/executor/subgraph_executor.cc b/ge/hybrid/executor/subgraph_executor.cc
index c4d866a9..f8f122b1 100644
--- a/ge/hybrid/executor/subgraph_executor.cc
+++ b/ge/hybrid/executor/subgraph_executor.cc
@@ -231,16 +231,16 @@ Status SubgraphExecutor::PrepareNodes() {
         } else {
           node_state->SetKernelTask(node_item.kernel_task);
         }
+        auto unique_task_context = TaskContext::Create(*node_state->GetNodeItem(), context_, subgraph_context_.get());
+        GE_CHECK_NOTNULL(unique_task_context);
+        const auto &task = node_state->GetKernelTask();
+        if (task == nullptr) {
+          GELOGE(INTERNAL_ERROR, "[%s] NodeTask is null.", node_state->GetName().c_str());
+          return INTERNAL_ERROR;
+        }
+        auto shared_task_context = std::shared_ptr<TaskContext>(unique_task_context.release());
+        node_state->SetTaskContext(shared_task_context);
       }
-      auto unique_task_context = TaskContext::Create(*node_state->GetNodeItem(), context_, subgraph_context_.get());
-      GE_CHECK_NOTNULL(unique_task_context);
-      const auto &task = node_state->GetKernelTask();
-      if (task == nullptr) {
-        GELOGE(INTERNAL_ERROR, "[%s] NodeTask is null.", node_state->GetName().c_str());
-        return INTERNAL_ERROR;
-      }
-      auto shared_task_context = std::shared_ptr<TaskContext>(unique_task_context.release());
-      node_state->SetTaskContext(shared_task_context);
     }
 
     if (!ready_queue_.Push(p_node_state)) {