From 06e6f353480200397f40c7abd72e106ea2f7d792 Mon Sep 17 00:00:00 2001
From: hwjiaorui <jiaorui3@huawei.com>
Date: Fri, 7 Jan 2022 17:10:08 +0800
Subject: [PATCH] fix compile shape

---
 mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc b/mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc
index 57fe68f92bc..9c26200fd4e 100644
--- a/mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc
+++ b/mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc
@@ -869,7 +869,7 @@ std::vector<int64_t> AnfRuntimeAlgorithm::GetInputDeviceShapeForTbeBuild(const A
     infer_shape = trans::PaddingShape(infer_shape, format, GetInputReshapeType(node, input_idx));
   }
   auto dtype = GetInputDeviceDataType(node, input_idx);
-  return trans::TransShapeToDevice(infer_shape, format, node, input_idx, dtype);
+  return trans::TransShapeToDevice(infer_shape, format, node, input_idx, dtype, false);
 }
 
 std::vector<size_t> AnfRuntimeAlgorithm::GetInputDeviceShape(const AnfNodePtr &node, size_t input_idx) {