From 76f341221eaf615741a0c8f14810e41977fd765c Mon Sep 17 00:00:00 2001
From: buxue <zhangbuxue@huawei.com>
Date: Tue, 22 Sep 2020 17:28:27 +0800
Subject: [PATCH] choose ai_core when input is 1D Tensor

---
 .../tbe/tbe_kernel_select/tbe_property_checker.cc               | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
index e57370bf85..bd6e114004 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
@@ -63,7 +63,7 @@ static bool CheckStridedSlice(const CNodePtr &cnode) {
                         << input->ToString();
     }
     int base_number = 2;
-    if (shrink_axis_mask >= std::pow<int, int>(base_number, input_dims - 1)) {
+    if (shrink_axis_mask >= std::pow<int, int>(base_number, input_dims - 1) && input_dims > 1) {
       return false;
     }
   }