From ad2ce8a591aa577840e0251f96331607f0c83ed6 Mon Sep 17 00:00:00 2001
From: yuchaojie <yuchaojie1@huawei.com>
Date: Tue, 6 Jul 2021 17:17:21 +0800
Subject: [PATCH] fix allgather op select bug

---
 .../ccsrc/backend/kernel_compiler/hccl/hccl_kernel_metadata.cc  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/hccl/hccl_kernel_metadata.cc b/mindspore/ccsrc/backend/kernel_compiler/hccl/hccl_kernel_metadata.cc
index 92d26431f8e..dfe141ce190 100755
--- a/mindspore/ccsrc/backend/kernel_compiler/hccl/hccl_kernel_metadata.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/hccl/hccl_kernel_metadata.cc
@@ -45,7 +45,7 @@ std::string GetKernelFormat(const CNodePtr &kernel_node, size_t index) {
     return format;
   }
   auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, index);
-  if (op_name == kAllGatherOpName) {
+  if (op_name == kAllGatherOpName && input_shape.size() <= kShape4dDims) {
     auto pad_shape = trans::PaddingShapeTo4dDefault(input_shape);
     if (pad_shape[N_nchw] % kCubeSize != 0 || pad_shape[C_nchw] % kCubeSize != 0) {
       return kOpFormat_DEFAULT;