Browse Source

GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

pull/2008/head
zhujingjing 3 years ago
parent
commit
74c9f1268c
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      ge/hybrid/node_executor/aicore/aicore_op_task.cc

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

@@ -571,8 +571,7 @@ Status AiCoreOpTask::UpdateArgs(TaskContext &task_context) {

if (shape_buffer_ != nullptr) {
arg_base_[index++] = reinterpret_cast<uintptr_t>(shape_buffer_->GetData());
GELOGD("Node:%s add shape buffer addr to args.",
task_context.GetNodeName());
GELOGD("Node:%s add shape buffer addr to args.", task_context.GetNodeName());
}

int workspace_num = static_cast<int>(task_context.NumWorkspaces());


Loading…
Cancel
Save