Merge pull request #1355 from yingcong-wu/yc/demangle-kernelname

[DeviceSanitizerLayer] Demangle the kernel name in error reporting

Author: pbalcer

source/loader/layers/sanitizer/asan_interceptor.cpp

@@ -281,6 +281,9 @@ void SanitizerInterceptor::postLaunchKernel(ur_kernel_handle_t Kernel,
         const char *Func = AH->Func[0] ? AH->Func : "<unknown func>";
         auto KernelName = getKernelName(Kernel);
 
+        // Try to demangle the kernel name
+        KernelName = DemangleName(KernelName);
+
         context.logger.always("\n====ERROR: DeviceSanitizer: {} on {}",
                               DeviceSanitizerFormat(AH->ErrorType),
                               DeviceSanitizerFormat(AH->MemoryType));

source/loader/layers/sanitizer/common.hpp

@@ -17,6 +17,7 @@
 
 #include <cassert>
 #include <cstdint>
+#include <string>
 
 namespace ur_sanitizer_layer {
 
@@ -107,4 +108,6 @@ bool DestroyShadowMem();
 
 void *GetMemFunctionPointer(const char *);
 
+std::string DemangleName(const std::string &name);
+
 } // namespace ur_sanitizer_layer

source/loader/layers/sanitizer/linux/san_utils.cpp

@@ -15,8 +15,10 @@
 #include "ur_sanitizer_layer.hpp"
 
 #include <asm/param.h>
+#include <cxxabi.h>
 #include <dlfcn.h>
 #include <gnu/lib-names.h>
+#include <string>
 #include <sys/mman.h>
 
 extern "C" __attribute__((weak)) void __asan_init(void);
@@ -84,4 +86,15 @@ void *GetMemFunctionPointer(const char *FuncName) {
     return ptr;
 }
 
+std::string DemangleName(const std::string &name) {
+    std::string result = name;
+    char *demangled =
+        abi::__cxa_demangle(name.c_str(), nullptr, nullptr, nullptr);
+    if (demangled) {
+        result = demangled;
+        free(demangled);
+    }
+    return result;
+}
+
 } // namespace ur_sanitizer_layer