From e22db35e1a05f13fe4dc2c9b2d4a8784ad666a66 Mon Sep 17 00:00:00 2001
From: jjfeing <jiangjianfei3@huawei.com>
Date: Tue, 12 Oct 2021 11:29:22 +0800
Subject: [PATCH] fix workspace error

---
 .../kernel_compiler/tbe/ascend_kernel_compile.cc       | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/ascend_kernel_compile.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/ascend_kernel_compile.cc
index db051d479c2..1002ac971e7 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/tbe/ascend_kernel_compile.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/ascend_kernel_compile.cc
@@ -606,14 +606,16 @@ bool AscendKernelCompileManager::AscendSingleOpCompile(const std::vector<AnfNode
     std::vector<size_t> in_size_list;
     std::vector<size_t> out_size_list;
     (void)TbeKernelBuild::GetIOSize(kernel_json, &in_size_list, &out_size_list, node);
-    if (!is_tune_flag_ && op_debug_level_ != "1" &&
-        build_manager_->SearchInCache(json_name, in_size_list, out_size_list, node.get())) {
-      continue;
-    }
+    // step1: if same node has been dispitch, no need to compile
     if (single_processed_kernels_.find(json_name) != single_processed_kernels_.end()) {
       build_manager_->SaveSameOpInfo(node, json_name, in_size_list, out_size_list);
       continue;
     }
+    // step2: if node has in the cache, load the cache.
+    if (!is_tune_flag_ && op_debug_level_ != "1" &&
+        build_manager_->SearchInCache(json_name, in_size_list, out_size_list, node.get())) {
+      continue;
+    }
     (void)single_processed_kernels_.insert(json_name);
 
     nlohmann::json build_json;