[CINN]Optimize cinn compile error report (#67738)

phlrain · web-flow · commit 122c3b5ca431 · 2024-08-27T18:05:56.000+08:00
* update

* update

* optimize cinn compile error
diff --git a/paddle/cinn/backends/nvrtc/nvrtc_util.cc b/paddle/cinn/backends/nvrtc/nvrtc_util.cc
@@ -187,9 +187,11 @@ std::string Compiler::CompileCudaSource(const std::string& code,
     std::string log;
     log.resize(log_size);
     NVRTC_CALL(nvrtcGetProgramLog(prog, &log[0]));
-    PADDLE_ENFORCE_EQ(compile_res,
-                      NVRTC_SUCCESS,
-                      ::common::errors::Fatal("NVRTC compilation failed"));
+
+    PADDLE_ENFORCE_EQ(
+        compile_res,
+        NVRTC_SUCCESS,
+        ::common::errors::Fatal("NVRTC compilation failed: %s", log));
   }
 
   size_t size;
diff --git a/paddle/cinn/ir/ir_base.cc b/paddle/cinn/ir/ir_base.cc
@@ -132,7 +132,7 @@ int32_t Expr::as_int32() const {
       true,
       phi::errors::InvalidArgument(
           "Invalid type. The type must be an 32-bit integer type. %s",
-          utils::enforce::GetCurrentTraceBackString()));
+          ::common::GetCurrentTraceBackString()));
   return As<IntImm>()->value;
 }
 int64_t Expr::as_int64() const {
diff --git a/paddle/cinn/utils/error.h b/paddle/cinn/utils/error.h
@@ -43,79 +43,6 @@
 namespace cinn {
 namespace utils {
 
-namespace enforce {
-
-#ifdef __GNUC__
-inline std::string demangle(std::string name) {
-  int status = -4;  // some arbitrary value to eliminate the compiler warning
-  std::unique_ptr<char, void (*)(void*)> res{
-      abi::__cxa_demangle(name.c_str(), NULL, NULL, &status), std::free};
-  return (status == 0) ? res.get() : name;
-}
-#else
-inline std::string demangle(std::string name) { return name; }
-#endif
-
-static std::string GetErrorSummaryString(const std::string& what,
-                                         const char* file,
-                                         int line) {
-  std::ostringstream sout;
-  sout << "\n----------------------\nError Message "
-          "Summary:\n----------------------\n";
-  sout << what << "(at " << file << " : " << line << ")" << std::endl;
-  return sout.str();
-}
-
-static std::string GetCurrentTraceBackString() {
-  std::ostringstream sout;
-  sout << "\n\n--------------------------------------\n";
-  sout << "C++ Traceback (most recent call last):";
-  sout << "\n--------------------------------------\n";
-#if !defined(_WIN32) && !defined(PADDLE_WITH_MUSL)
-  static constexpr int TRACE_STACK_LIMIT = 100;
-
-  void* call_stack[TRACE_STACK_LIMIT];
-  auto size = backtrace(call_stack, TRACE_STACK_LIMIT);
-  auto symbols = backtrace_symbols(call_stack, size);
-  Dl_info info;
-  int idx = 0;
-  int end_idx = 0;
-  for (int i = size - 1; i >= end_idx; --i) {
-    if (dladdr(call_stack[i], &info) && info.dli_sname) {
-      auto demangled = demangle(info.dli_sname);
-      std::string path(info.dli_fname);
-      // C++ traceback info are from core.so
-      if (path.substr(path.length() - 3).compare(".so") == 0) {
-        sout << idx++ << " " << demangled << "\n";
-      }
-    }
-  }
-  free(symbols);
-#else
-  sout << "Not support stack backtrace yet.\n";
-#endif
-  return sout.str();
-}
-
-static std::string GetTraceBackString(const std::string& what,
-                                      const char* file,
-                                      int line) {
-  return GetCurrentTraceBackString() + GetErrorSummaryString(what, file, line);
-}
-
-struct EnforceNotMet : public std::exception {
- public:
-  EnforceNotMet(const std::string& str, const char* file, int line)
-      : err_str_(GetTraceBackString(str, file, line)) {}
-
-  const char* what() const noexcept override { return err_str_.c_str(); }
-
- private:
-  std::string err_str_;
-};
-
-}  // namespace enforce
-
 /**
  *  \brief Indicates the level of printing error message in the current
  * operation
diff --git a/paddle/cinn/utils/multi_threading.cc b/paddle/cinn/utils/multi_threading.cc
@@ -87,20 +87,25 @@ void parallel_run(const WorkerFuncType& fn,
     int counter = worker(tid);
     VLOG(4) << "Thread-0  process " << counter << " tasks.";
 
-    for (auto&& future : futures) {
+    for (auto& future : futures) {
       counter = future.get();
       ++tid;
       VLOG(4) << "Thread-" << tid << " process " << counter << " tasks.";
     }
-  } catch (const std::exception& e) {
-    std::stringstream ss;
-    ss << "parallel_run incurs error: " << e.what();
-    PADDLE_THROW(::common::errors::Fatal(ss.str()));
-  }
 
-  // join threads
-  for (auto&& thread : threads) {
-    thread.join();
+    // join threads
+    for (auto&& thread : threads) {
+      thread.join();
+    }
+  } catch (::common::EnforceNotMet& ex) {
+    LOG(ERROR) << ex.error_str();
+    PADDLE_THROW(
+        ::common::errors::Fatal("Parallel compile Paddle enfore error"));
+  } catch (const std::exception& e) {
+    LOG(ERROR) << "Parallel compile error " << e.what();
+    PADDLE_THROW(::common::errors::Fatal("Parallel compile std::exception"));
+  } catch (...) {
+    PADDLE_THROW(::common::errors::Fatal("Parallel compile unknow exception"));
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -132,7 +132,7 @@ int32_t Expr::as_int32() const {`
`132`	`132`	`true,`
`133`	`133`	`phi::errors::InvalidArgument(`
`134`	`134`	`"Invalid type. The type must be an 32-bit integer type. %s",`
`135`		`- utils::enforce::GetCurrentTraceBackString()));`
	`135`	`+ ::common::GetCurrentTraceBackString()));`
`136`	`136`	`return As<IntImm>()->value;`
`137`	`137`	`}`
`138`	`138`	`int64_t Expr::as_int64() const {`