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) {