[CINN] fix accuracy_check_pass cannot find kernel name (#66791)

chen2016013 · web-flow · commit d7c5ce7ce583 · 2024-08-01T12:05:54.000+08:00
* fix

* Update compilation_cache.h
diff --git a/paddle/cinn/hlir/framework/pir/compilation_cache.h b/paddle/cinn/hlir/framework/pir/compilation_cache.h
@@ -49,6 +49,7 @@ class BackendResource final {
     return backend_compiler_;
   }
   pir::CINNKernelInfo GenerateKernelInfo() const;
+  const std::string& GetHostFuncName() const { return host_fn_name_; }
 
  private:
   std::string host_fn_name_;
@@ -69,6 +70,14 @@ class CompilationResult final {
     backend_resource_ = other;
   }
 
+  const std::string& GetHostFuncName() const {
+    PADDLE_ENFORCE_NOT_NULL(GetBackendResource(),
+                            ::common::errors::PreconditionNotMet(
+                                "Found backend_resource_ is nullptr, please "
+                                "call SetBackendResource first."));
+    return GetBackendResource()->GetHostFuncName();
+  }
+
   pir::CINNKernelInfo GetKernelInfo() {
     PADDLE_ENFORCE_NOT_NULL(backend_resource_,
                             ::common::errors::PreconditionNotMet(
diff --git a/paddle/cinn/hlir/framework/pir_compiler.cc b/paddle/cinn/hlir/framework/pir_compiler.cc
@@ -156,7 +156,7 @@ void CompilationContextMapper::Construct(
 
   for (size_t i = 0; i < groups.size(); ++i) {
     fusion_infos_.emplace_back(*groups[i]);
-    VLOG(5) << "Construct FusionInfo: " << fusion_infos_[i]
+    VLOG(4) << "Construct FusionInfo: " << fusion_infos_[i]
             << " for group: " << *groups[i];
     // If FLAGS_enable_cinn_compile_cache=False, Cache strategy will not take
     // effects.
@@ -206,8 +206,10 @@ void CompilationContextMapper::UpdateGlobalCache() {
                       ::common::errors::PreconditionNotMet(
                           "Required mapper_index < fusion_infos_.size()."));
     const auto& fusion_info = fusion_infos_[mapper_index_[i]];
-    VLOG(5) << "Insert new compiled result into cache, fusion_info: "
-            << fusion_info;
+    VLOG(4) << "============== Insert new compiled result into cache, "
+               "fusion_info: ==============\n"
+            << fusion_info << ", host func name: "
+            << compilation_results_[i]->GetHostFuncName();
     CompilationCache::Instance().Insert(fusion_info, compilation_results_[i]);
   }
 }
