Browse Source

fix aclInferShapeAndType

tags/v1.1.0
dongduo 4 years ago
parent
commit
737a616ec4
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      ge/ir_build/ge_ir_build.cc

+ 6
- 0
ge/ir_build/ge_ir_build.cc View File

@@ -442,6 +442,12 @@ graphStatus aclgrphInferShapeAndType(ge::Graph &graph) {
auto compute_graph = GraphUtils::GetComputeGraph(graph); auto compute_graph = GraphUtils::GetComputeGraph(graph);
GE_CHECK_NOTNULL(compute_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()) { for (auto &node: compute_graph->GetAllNodes()) {
graphStatus ret = ShapeRefiner::InferShapeAndType(node); graphStatus ret = ShapeRefiner::InferShapeAndType(node);
if (ret == GRAPH_PARAM_INVALID) { if (ret == GRAPH_PARAM_INVALID) {


Loading…
Cancel
Save