SWDEV-299162 - Add new comgr API to demangle names

Change-Id: I84174d7e40e9afff9a8ea135abadb69d580478e0


[ROCm/clr commit: b5a57327bb]

projects/clr/rocclr/device/comgrctx.cpp

@@ -85,6 +85,7 @@ bool Comgr::LoadLib() {
   GET_COMGR_SYMBOL(amd_comgr_iterate_symbols)
   GET_COMGR_SYMBOL(amd_comgr_symbol_lookup)
   GET_COMGR_SYMBOL(amd_comgr_symbol_get_info)
+  GET_COMGR_SYMBOL(amd_comgr_demangle_symbol_name)
   is_ready_ = true;
   return true;
 }

projects/clr/rocclr/device/comgrctx.hpp

@@ -71,6 +71,7 @@ typedef amd_comgr_status_t (*t_amd_comgr_index_list_metadata)(amd_comgr_metadata
 typedef amd_comgr_status_t (*t_amd_comgr_iterate_symbols)(amd_comgr_data_t data, amd_comgr_status_t(*callback)(amd_comgr_symbol_t symbol, void *user_data), void *user_data);
 typedef amd_comgr_status_t (*t_amd_comgr_symbol_lookup)(amd_comgr_data_t data, const char *name, amd_comgr_symbol_t *symbol);
 typedef amd_comgr_status_t (*t_amd_comgr_symbol_get_info)(amd_comgr_symbol_t symbol, amd_comgr_symbol_info_t attribute, void *value);
+typedef amd_comgr_status_t (*t_amd_comgr_demangle_symbol_name)(amd_comgr_data_t MangledSymbolName, amd_comgr_data_t* DemangledSymbolName);
 
 struct ComgrEntryPoints {
   void* handle;
@@ -119,6 +120,7 @@ struct ComgrEntryPoints {
   t_amd_comgr_iterate_symbols           amd_comgr_iterate_symbols;
   t_amd_comgr_symbol_lookup             amd_comgr_symbol_lookup;
   t_amd_comgr_symbol_get_info           amd_comgr_symbol_get_info;
+  t_amd_comgr_demangle_symbol_name      amd_comgr_demangle_symbol_name;
 };
 
 #ifdef COMGR_DYN_DLL
@@ -274,8 +276,12 @@ public:
   static amd_comgr_status_t symbol_get_info(amd_comgr_symbol_t symbol, amd_comgr_symbol_info_t attribute, void *value) {
     return COMGR_DYN(amd_comgr_symbol_get_info)(symbol, attribute, value);
   }
+  static amd_comgr_status_t demangle_symbol_name(amd_comgr_data_t MangledSymbolName,
+                                                 amd_comgr_data_t* DemangledSymbolName) {
+    return COMGR_DYN(amd_comgr_demangle_symbol_name)(MangledSymbolName, DemangledSymbolName);
+  }
 
-private:
+ private:
   static ComgrEntryPoints cep_;
   static bool is_ready_;
 };

projects/clr/rocclr/device/devprogram.cpp

@@ -2894,6 +2894,51 @@ const bool Program::getLoweredNames(std::vector<std::string>* mangledNames) cons
 #endif
 }
 
+bool Program::getDemangledName(const std::string& mangledName, std::string& demangledName) const {
+#if defined(USE_COMGR_LIBRARY)
+  amd_comgr_data_t mangled_data;
+  amd_comgr_data_t demangled_data;
+
+  if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::create_data(AMD_COMGR_DATA_KIND_BYTES, &mangled_data))
+    return false;
+
+  if (AMD_COMGR_STATUS_SUCCESS !=
+      amd::Comgr::set_data(mangled_data, mangledName.size(), mangledName.c_str())) {
+    amd::Comgr::release_data(mangled_data);
+    return false;
+  }
+
+  if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::demangle_symbol_name(mangled_data, &demangled_data)) {
+    amd::Comgr::release_data(mangled_data);
+    return false;
+  }
+
+  size_t demangled_size = 0;
+  if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::get_data(demangled_data, &demangled_size, NULL)) {
+    amd::Comgr::release_data(mangled_data);
+    amd::Comgr::release_data(demangled_data);
+    return false;
+  }
+
+  demangledName.resize(demangled_size);
+
+  if (AMD_COMGR_STATUS_SUCCESS !=
+      amd::Comgr::get_data(demangled_data, &demangled_size,
+                           const_cast<char*>(demangledName.data()))) {
+    amd::Comgr::release_data(mangled_data);
+    amd::Comgr::release_data(demangled_data);
+    return false;
+  }
+
+  amd::Comgr::release_data(mangled_data);
+  amd::Comgr::release_data(demangled_data);
+  return true;
+#else
+  assert(!"No COMGR loaded");
+  return false;
+#endif
+}
+
 bool Program::getGlobalFuncFromCodeObj(std::vector<std::string>* func_names) const {
 #if defined(USE_COMGR_LIBRARY)
   return getSymbolsFromCodeObj(func_names, AMD_COMGR_SYMBOL_TYPE_FUNC);

projects/clr/rocclr/device/devprogram.hpp

@@ -277,6 +277,9 @@ class Program : public amd::HeapObject {
   //! Get mangled name of a name expresion
   const bool getLoweredNames(std::vector<std::string>* mangledNames) const;
 
+  //! Get demangled names
+  bool getDemangledName(const std::string& mangledNames, std::string& demangledNames) const;
+
   bool getGlobalFuncFromCodeObj(std::vector<std::string>* func_names) const;
   bool getGlobalVarFromCodeObj(std::vector<std::string>* var_names) const;
   bool getUndefinedVarFromCodeObj(std::vector<std::string>* var_names) const;