diff --git a/ge/hybrid/node_executor/aicore/aicore_node_executor.cc b/ge/hybrid/node_executor/aicore/aicore_node_executor.cc index 93458cfe..01bd9717 100755 --- a/ge/hybrid/node_executor/aicore/aicore_node_executor.cc +++ b/ge/hybrid/node_executor/aicore/aicore_node_executor.cc @@ -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());