From 29be15b8e21cf5e598769d1bc00107af02f6e0f6 Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Mon, 4 Jan 2021 11:01:01 +0800
Subject: [PATCH] bugfix

---
 ge/ir_build/ge_ir_build.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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;
   }