Browse Source

GE support aicore DEPEND_SHAPE_RANGE op update outputs shape

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

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

@@ -223,6 +223,7 @@ Status AiCoreNodeTask::ExecuteAsync(TaskContext &context, std::function<void()>
auto callback = [=, &context]() {
Status callback_ret = SUCCESS;
if (!tasks_.empyt()) {
// only last task need update outputs shape
auto task = tasks_.back().get();
if (task->GetUnknownShapeOpType() == DEPEND_SHAPE_RANGE) {
GELOGD("Node[%s] need update outputs shape.", context.GetNodeName());


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

@@ -55,7 +55,7 @@ bool TbeHandleRegistry::AddHandle(std::unique_ptr<TbeHandleHolder> &&holder) {

Status AiCoreOpTask::Init(const OpDesc &op_desc,
const domi::TaskDef &task_def) {
GE_CHK_STATUS_RET_NOLOG(AiCoreOpTask::DoInit(op_desc, task_def));
GE_CHK_STATUS_RET_NOLOG(DoInit(op_desc, task_def));
int32_t unknown_shape_op_type_val = static_cast<int32_t>(DEPEND_IN_SHAPE);
(void)AttrUtils::GetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE,
unknown_shape_op_type_val);


Loading…
Cancel
Save