From 737a616ec431672ef5d45ef0f1d20fc63ee40dc5 Mon Sep 17 00:00:00 2001 From: dongduo Date: Fri, 30 Oct 2020 19:52:36 +0800 Subject: [PATCH] fix aclInferShapeAndType --- ge/ir_build/ge_ir_build.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc index 0fd613ed..b845bd75 100644 --- a/ge/ir_build/ge_ir_build.cc +++ b/ge/ir_build/ge_ir_build.cc @@ -442,6 +442,12 @@ graphStatus aclgrphInferShapeAndType(ge::Graph &graph) { auto compute_graph = GraphUtils::GetComputeGraph(graph); GE_CHECK_NOTNULL(compute_graph); + auto ret = compute_graph->InferOriginFormat(); + if (ret != GRAPH_SUCCESS) { + GELOGE(ret, "Acl InferOriginFormat failed."); + return ret; + } + for (auto &node: compute_graph->GetAllNodes()) { graphStatus ret = ShapeRefiner::InferShapeAndType(node); if (ret == GRAPH_PARAM_INVALID) {