Runtime Initialization Tracing (#1105)

* Runtime initialization tracing

- calbacks and buffer entries notifying when a runtime has been initialized

* Minor cleanup to registration.cpp

* JSON tool implementation

* Increase perfetto_reader timeout

* Handle perfetto_reader timeout when attr doesn't exist

* clang-tidy fixes to memory_allocation.cpp

source/lib/rocprofiler-sdk/registration.cpp

@@ -45,6 +45,7 @@
 #include "lib/rocprofiler-sdk/pc_sampling/code_object.hpp"
 #include "lib/rocprofiler-sdk/pc_sampling/service.hpp"
 #include "lib/rocprofiler-sdk/rccl/rccl.hpp"
+#include "lib/rocprofiler-sdk/runtime_initialization.hpp"
 
 #include <rocprofiler-sdk/context.h>
 #include <rocprofiler-sdk/fwd.h>
@@ -52,6 +53,9 @@
 #include <rocprofiler-sdk/marker.h>
 #include <rocprofiler-sdk/version.h>
 
+#include <hsa/hsa_api_trace.h>
+#include <hip/amd_detail/hip_api_trace.hpp>
+
 #include <fmt/format.h>
 
 #include <dlfcn.h>
@@ -59,9 +63,9 @@
 #include <unistd.h>
 #include <atomic>
 #include <cctype>
+#include <cstddef>
 #include <cstdint>
 #include <fstream>
-#include <hip/amd_detail/hip_api_trace.hpp>
 #include <iostream>
 #include <limits>
 #include <memory>
@@ -732,6 +736,11 @@ rocprofiler_set_api_table(const char* name,
         // install rocprofiler API wrappers
         rocprofiler::hip::update_table(hip_runtime_api_table);
 
+        // Tracing notifications the runtime has initialized
+        rocprofiler::runtime_init::initialize(
+            ROCPROFILER_RUNTIME_INITIALIZATION_HIP, lib_version, lib_instance);
+
+        // allow tools to install API wrappers
         rocprofiler::intercept_table::notify_intercept_table_registration(
             ROCPROFILER_HIP_RUNTIME_TABLE,
             lib_version,
@@ -754,6 +763,7 @@ rocprofiler_set_api_table(const char* name,
         // install rocprofiler API wrappers
         rocprofiler::hip::update_table(hip_compiler_api_table);
 
+        // allow tools to install API wrappers
         rocprofiler::intercept_table::notify_intercept_table_registration(
             ROCPROFILER_HIP_COMPILER_TABLE,
             lib_version,
@@ -773,6 +783,12 @@ rocprofiler_set_api_table(const char* name,
 
         auto* hsa_api_table = static_cast<HsaApiTable*>(*tables);
 
+#if ROCPROFILER_SDK_HSA_PC_SAMPLING > 0
+        auto hsa_api_table_size = hsa_api_table->version.minor_id;
+        auto runtime_pc_sampling_table =
+            (offsetof(::HsaApiTable, pc_sampling_ext_) < hsa_api_table_size);
+#endif
+
         // store a reference of the HsaApiTable implementations for invoking these functions
         // without going through tracing wrappers
         rocprofiler::hsa::copy_table(hsa_api_table->core_, lib_instance);
@@ -781,7 +797,8 @@ rocprofiler_set_api_table(const char* name,
         rocprofiler::hsa::copy_table(hsa_api_table->finalizer_ext_, lib_instance);
         rocprofiler::hsa::copy_table(hsa_api_table->tools_, lib_instance);
 #if ROCPROFILER_SDK_HSA_PC_SAMPLING > 0
-        rocprofiler::hsa::copy_table(hsa_api_table->pc_sampling_ext_, lib_instance);
+        if(runtime_pc_sampling_table)
+            rocprofiler::hsa::copy_table(hsa_api_table->pc_sampling_ext_, lib_instance);
 #endif
 
         // need to construct agent mappings before initializing the queue controller
@@ -796,7 +813,8 @@ rocprofiler_set_api_table(const char* name,
         rocprofiler::code_object::initialize(hsa_api_table);
         rocprofiler::thread_trace::initialize(hsa_api_table);
 #if ROCPROFILER_SDK_HSA_PC_SAMPLING > 0
-        rocprofiler::pc_sampling::code_object::initialize(hsa_api_table);
+        if(runtime_pc_sampling_table)
+            rocprofiler::pc_sampling::code_object::initialize(hsa_api_table);
 #endif
 
         // install rocprofiler API wrappers
@@ -808,9 +826,14 @@ rocprofiler_set_api_table(const char* name,
 
 #if ROCPROFILER_SDK_HSA_PC_SAMPLING > 0
         // Initialize PC sampling service if configured
-        rocprofiler::pc_sampling::post_hsa_init_start_active_service();
+        if(runtime_pc_sampling_table)
+            rocprofiler::pc_sampling::post_hsa_init_start_active_service();
 #endif
 
+        // Tracing notifications the runtime has initialized
+        rocprofiler::runtime_init::initialize(
+            ROCPROFILER_RUNTIME_INITIALIZATION_HSA, lib_version, lib_instance);
+
         // allow tools to install API wrappers
         rocprofiler::intercept_table::notify_intercept_table_registration(
             ROCPROFILER_HSA_TABLE, lib_version, lib_instance, std::make_tuple(hsa_api_table));
@@ -839,6 +862,11 @@ rocprofiler_set_api_table(const char* name,
         rocprofiler::marker::update_table(roctx_ctrl);
         rocprofiler::marker::update_table(roctx_name);
 
+        // Tracing notifications the runtime has initialized
+        rocprofiler::runtime_init::initialize(
+            ROCPROFILER_RUNTIME_INITIALIZATION_MARKER, lib_version, lib_instance);
+
+        // allow tools to install API wrappers
         rocprofiler::intercept_table::notify_intercept_table_registration(
             ROCPROFILER_MARKER_CORE_TABLE, lib_version, lib_instance, std::make_tuple(roctx_core));
 
@@ -867,6 +895,11 @@ rocprofiler_set_api_table(const char* name,
         // install rocprofiler API wrappers
         rocprofiler::rccl::update_table(rccl_api);
 
+        // Tracing notifications the runtime has initialized
+        rocprofiler::runtime_init::initialize(
+            ROCPROFILER_RUNTIME_INITIALIZATION_RCCL, lib_version, lib_instance);
+
+        // allow tools to install API wrappers
         rocprofiler::intercept_table::notify_intercept_table_registration(
             ROCPROFILER_RCCL_TABLE, lib_version, lib_instance, std::make_tuple(rccl_api));
     }