SWDEV-516846: Fix serialization services conflicts and ATT counter streaming (#230)

* Update TT API * Rework serialization * update att_core * Fix tests * Fix tool * Formatting * Fix perfcounter * Formatting * Rename agent TT * Format * Workaround for codeQL alert * Tidy fix * Fix compiler error * Tidy * Fix some tests * Fixing some tests * formatting * Fixing ATT serialization * Format * Fix test commandline * Fixing init order * Format * Tidy fixes * Removing unused sample * Fix tests and schema * Added ATT + PMC test * Fix mode * Fix file mode * Review comments * Fix typo * Review comments * Review comments * Fix missing id inc after review comment * Review comments * Suggested Fixes * Testing changes * Test fix * Build fixes * Minor build fix --------- Co-authored-by: Giovanni Baraldi <gbaraldi@amd.com> Co-authored-by: Benjamin Welton <bewelton@amd.com> Co-authored-by: Welton, Benjamin <Benjamin.Welton@amd.com>
2025-03-15 02:11:10 +01:00
@@ -83,10 +83,36 @@ tool_init(rocprofiler_client_finalize_t /* fini_func */, void* tool_data)
                                                       tool_data),
        "code object tracing service configure");

+    std::vector<rocprofiler_agent_id_t> agents{};
+
    ROCPROFILER_CALL(
-        rocprofiler_configure_dispatch_thread_trace_service(
-            client_ctx, nullptr, 0, dispatch_callback, Callbacks::shader_data_callback, tool_data),
-        "thread trace service configure");
+        rocprofiler_query_available_agents(
+            ROCPROFILER_AGENT_INFO_VERSION_0,
+            [](rocprofiler_agent_version_t, const void** _agents, size_t _num_agents, void* _data) {
+                auto* agent_v = static_cast<std::vector<rocprofiler_agent_id_t>*>(_data);
+                for(size_t i = 0; i < _num_agents; ++i)
+                {
+                    auto* agent = static_cast<const rocprofiler_agent_v0_t*>(_agents[i]);
+                    if(agent->type == ROCPROFILER_AGENT_TYPE_GPU) agent_v->emplace_back(agent->id);
+                }
+                return ROCPROFILER_STATUS_SUCCESS;
+            },
+            sizeof(rocprofiler_agent_v0_t),
+            &agents),
+        "Failed to iterate agents");
+
+    for(auto id : agents)
+    {
+        ROCPROFILER_CALL(
+            rocprofiler_configure_dispatch_thread_trace_service(client_ctx,
+                                                                id,
+                                                                nullptr,
+                                                                0,
+                                                                dispatch_callback,
+                                                                Callbacks::shader_data_callback,
+                                                                tool_data),
+            "thread trace service configure");
+    }

    int valid_ctx = 0;
    ROCPROFILER_CALL(rocprofiler_context_is_valid(client_ctx, &valid_ctx),