bugfix

4 years ago · 29be15b8e2
parent df0a364762
commit 29be15b8e2
1 changed files with 1 additions and 1 deletions
--- a/ge/ir_build/ge_ir_build.cc
+++ b/ge/ir_build/ge_ir_build.cc
@ -611,7 +611,7 @@ graphStatus aclgrphInferShapeAndType(ge::Graph &graph) {
    return GRAPH_PARAM_INVALID;
  }

-  auto ret = Impl::InferShapePrepare(root_graph);
+  auto ret = Impl::InferShapePrepare(compute_graph);
  if (ret != GRAPH_SUCCESS) {
    return ret;
  }