Browse Source

!1192 Bugfix

From: @xchu42
Reviewed-by: @ji_chen,@wqtshg
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
a7812f4bb3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/aicore/aicore_node_executor.cc

+ 1
- 1
ge/hybrid/node_executor/aicore/aicore_node_executor.cc View File

@@ -141,7 +141,7 @@ Status AiCoreNodeExecutor::CompileTask(const HybridModel &model,
auto node_key = std::to_string(model.GetModelId()) + "/" + shape_key;
GELOGD("NodeKey for %s = %s", node->GetName().c_str(), node_key.c_str());
auto aicore_task = registry.GetTask(node_key);
if (task != nullptr) {
if (aicore_task != nullptr) {
// The workspaces needed by a operator may differ with different shapes
op_desc->SetWorkspaceBytes(aicore_task->GetWorkspaceSizes());
GELOGI("AiCoreNodeExecutor(%s) CompileTask Skip.", node->GetName().c_str());


Loading…
Cancel
Save