Browse Source

SetTaskContext for known

tags/v1.3.0
zhangxiaokun 3 years ago
parent
commit
3e8f3470a3
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/hybrid/executor/subgraph_executor.cc

+ 1
- 0
ge/hybrid/executor/subgraph_executor.cc View File

@@ -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));


Loading…
Cancel
Save