diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc
index 95fb6749..8423c8bb 100644
--- 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;
   }