AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Search for rocprofiler_set_api_table(...) via RTLD_DEFAULT (#56)

Browse Source 
* Search for rocprofiler_set_api_table(...) via RTLD_DEFAULT

- check if rocprofiler_set_api_table function has already been loaded via dlsym(RTLD_DEFAULT, ...)
- fix warning about conflicting ROCM_DEP_ROCMCORE and ROCPROFILER_REGISTER_DEP_ROCMCORE

* Bump version to 0.4.0
This commit is contained in:
Jonathan R. Madsen
2024-02-25 21:14:27 -06:00
committed by GitHub
parent 1f0d06b6cb
commit e2db37ca8d
3 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/rocprofiler-register/rocprofiler_register.cpp
+12
View File
@@ -273,6 +273,18 @@ rocp_load_rocprofiler_lib(std::string _rocp_reg_lib)
if(rocprofiler_set_api_table) rocprofiler_lib_config_fn = &rocprofiler_set_api_table;
// return if found via LD_PRELOAD
if(rocprofiler_lib_config_fn)
return std::make_tuple(rocprofiler_lib_handle, rocprofiler_lib_config_fn);
// look to see if entrypoint function is already a symbol
*(void**) (&rocprofiler_lib_config_fn) =
dlsym(RTLD_DEFAULT, rocprofiler_lib_register_entrypoint);
// return if found via RTLD_DEFAULT
if(rocprofiler_lib_config_fn)
return std::make_tuple(rocprofiler_lib_handle, rocprofiler_lib_config_fn);
if(_rocp_reg_lib.empty()) _rocp_reg_lib = rocprofiler_lib_name;
auto _rocp_reg_lib_path = fs::path{ _rocp_reg_lib };