From d6b7ee64b7bc85ca35ee044f577d9557cc8e576f Mon Sep 17 00:00:00 2001
From: simson <wudengsong@huawei.com>
Date: Sun, 7 Feb 2021 17:48:37 +0800
Subject: [PATCH] fix error of AICPU so name

---
 .../backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc b/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc
index d11f278cdc..8b5f8aecaf 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc
@@ -79,7 +79,9 @@ void AicpuOpKernelMod::CreateCpuKernelInfo(const std::vector<AddressPtr> &inputs
     node_so_ = AICPU_OPS_SO_NAME;
     node_name_ = kCustRunApi;
   } else {
-    node_so_ = AICPU_OPS_SO_NAME;
+    if (node_so_ != CUST_AICPU_OPS_SO_NAME) {
+      node_so_ = AICPU_OPS_SO_NAME;
+    }
   }
   // InputOutputAddr
   vector<void *> io_addrs;
@@ -171,7 +173,9 @@ std::vector<TaskInfoPtr> AicpuOpKernelMod::GenTask(const std::vector<AddressPtr>
     node_so_ = AICPU_OPS_SO_NAME;
     node_name_ = kCustRunApi;
   } else {
-    node_so_ = AICPU_OPS_SO_NAME;
+    if (node_so_ != CUST_AICPU_OPS_SO_NAME) {
+      node_so_ = AICPU_OPS_SO_NAME;
+    }
   }
   std::vector<void *> input_data_addrs;
   (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs),