From a384ef05c577fda5558858b0b14ef83f122dfe33 Mon Sep 17 00:00:00 2001 From: zhujingjing Date: Thu, 15 Jul 2021 11:27:09 +0800 Subject: [PATCH] GE support aicore DEPEND_SHAPE_RANGE op update outputs shape --- ge/hybrid/node_executor/aicore/aicore_node_executor.cc | 1 + ge/hybrid/node_executor/aicore/aicore_op_task.cc | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/ge/hybrid/node_executor/aicore/aicore_node_executor.cc b/ge/hybrid/node_executor/aicore/aicore_node_executor.cc index 1f95d2ce..5f912a89 100755 --- a/ge/hybrid/node_executor/aicore/aicore_node_executor.cc +++ b/ge/hybrid/node_executor/aicore/aicore_node_executor.cc @@ -223,6 +223,7 @@ Status AiCoreNodeTask::ExecuteAsync(TaskContext &context, std::function 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()); diff --git a/ge/hybrid/node_executor/aicore/aicore_op_task.cc b/ge/hybrid/node_executor/aicore/aicore_op_task.cc index ec98b726..d2f45fa0 100644 --- a/ge/hybrid/node_executor/aicore/aicore_op_task.cc +++ b/ge/hybrid/node_executor/aicore/aicore_op_task.cc @@ -55,7 +55,7 @@ bool TbeHandleRegistry::AddHandle(std::unique_ptr &&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(DEPEND_IN_SHAPE); (void)AttrUtils::GetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, unknown_shape_op_type_val);