Browse Source

fix workspace

tags/v1.2.0
chuxing 4 years ago
parent
commit
1a90d7be43
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