Browse Source

GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

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

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

@@ -81,8 +81,7 @@ Status AiCoreOpTask::Init(const OpDesc &op_desc, const domi::TaskDef &task_def)
return SUCCESS;
}

Status AiCoreOpTask::DoInit(const OpDesc &op_desc,
const domi::TaskDef &task_def) {
Status AiCoreOpTask::DoInit(const OpDesc &op_desc, const domi::TaskDef &task_def) {
op_type_ = op_desc.GetType();
log_name_ = op_desc.GetName() + "_tvmbin";
log_id_ = log_id++;


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

@@ -111,8 +111,7 @@ class AiCoreOpTask {
Status RegisterKernelHandle(const OpDesc &op_desc);
Status InitWithKernelDef(const OpDesc &op_desc, const domi::TaskDef &task_def);
Status InitWithKernelDefWithHandle(const OpDesc &node, const domi::TaskDef &task_def);
Status UpdateShapeToOutputDesc(TaskContext &context, const GeShape &shape,
const int output_index) const;
Status UpdateShapeToOutputDesc(TaskContext &context, const GeShape &shape, const int output_index) const;

std::string stub_name_;
void *stub_func_ = nullptr;


Loading…
Cancel
Save