Return true on omnitrace_is_preloaded using a second environment vari…

…able
ROCm · Oct 4, 2024 · dd10a30 · dd10a30
1 parent c022bff
commit dd10a30
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/source/lib/omnitrace-dl/dl.cpp b/source/lib/omnitrace-dl/dl.cpp
@@ -108,12 +108,19 @@ get_omnitrace_dl_env()
                : get_env("OMNITRACE_DL_VERBOSE", get_omnitrace_env());
 }
 
-inline bool&
+inline bool
 get_omnitrace_is_preloaded()
 {
+    auto preloaded_env = get_env("OMNITRACE_DL_IS_PRELOADED", std::string{});
+    if(preloaded_env == "true")
+        return true;
+
     static bool _v = []() {
         auto&& _preload_libs = get_env("LD_PRELOAD", std::string{});
-        return (_preload_libs.find("libomnitrace-dl.so") != std::string::npos);
+        auto&& preloaded_f = _preload_libs.find("libomnitrace-dl.so") != std::string::npos;
+        if(preloaded_f)
+            setenv("OMNITRACE_DL_IS_PRELOADED", "true", 1);
+        return (preloaded_f);
     }();
     return _v;
 }