Initial Commit

Contributors:
Ammar ELWazir <aelwazir@amd.com>
AravindanC <aravindan.cheruvally@amd.com>
Benjamin Welton <bewelton@amd.com>
Ma, Bing <Bing.Ma@amd.com>
Chun Yang <chun.yang@amd.com>
Cole Nelson <cole.nelson@amd.com>
Ethan Stewart <ethan.stewart@amd.com>
Evgeny <evgeny.shcherbakov@amd.com>
Freddy Paul <Freddy.paul@amd.com>
Giovanni Baraldi <gbaraldi@amd.com>
Gopesh Bhardwaj <Gopesh.Bhardwaj@amd.com>
Icarus Sparry <icarus.sparry@amd.com>
itrowbri <Ian.Trowbridge@amd.com>
James Edwards <JamesAdrian.Edwards@amd.com>
jatang <jatang@amd.com>
Jeremy Newton <Jeremy.Newton@amd.com>
Jonathan Kim <jonathan.kim@amd.com>
Kent Russell <kent.russell@amd.com>
Kiumars Sabeti <kiumars.sabeti@amd.com>
Lang Yu <lang.yu@amd.com>
Laurent Morichetti <laurent.morichetti@amd.com>
Mallya, Ameya Keshava <AmeyaKeshava.Mallya@amd.com>
Manjunath Jakaraddi <manjunath.jakaraddi@amd.com>
Mark Laws <markdavid.laws@amd.com>
Mohan Kumar Mithur <Mohan.KumarMithur@amd.com>
Nicholas Curtis <nicurtis@amd.com>
Nirmal Unnikrishnan <Nirmal.Unnikrishnan@amd.com>
Parag Bhandari <parag.bhandari@amd.com>
Ranjith Ramakrishnan <Ranjith.Ramakrishnan@amd.com>
Robert Gregory <Robert.Gregory@amd.com>
Saravanan Solaiyappan <saravanan.solaiyappan@amd.com>
Saurabh Verma <saurabh.verma@amd.com>
Srihari Uttanur <srihari.u@amd.com>
Srinivasan Subramanian <srinivasan.subramanian@amd.com>
Sriraksha Nagaraj <Sriraksha.Nagaraj@amd.com>
Sushma Vaddireddy <svaddire@amd.com>
Xianwei Zhang <Xianwei.Zhang@amd.com>

src/core/pm4_factory.cpp

@@ -0,0 +1,76 @@
+// MIT License
+//
+// Copyright (c) 2017-2025 Advanced Micro Devices, Inc.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+// THE SOFTWARE.
+
+#include "pm4_factory.h"
+
+#include <mutex>
+#include <shared_mutex>
+
+namespace aql_profile {
+namespace {
+struct locked_agent_cache {
+  std::shared_mutex mutex;
+  std::unordered_map<uint64_t, AgentInfo> cache;
+
+  void add(uint64_t& agent_id, const AgentInfo& agent_info) {
+    auto lock = std::unique_lock{mutex};
+    agent_id = cache.size();
+    cache[agent_id] = agent_info;
+  }
+
+  const AgentInfo* get(uint64_t agent_id) {
+    auto lock = std::shared_lock{mutex};
+    auto it = cache.find(agent_id);
+    if (it == cache.end()) return nullptr;
+    return &it->second;
+  }
+};
+
+locked_agent_cache& get_cache() {
+  static auto* cache = new locked_agent_cache{};
+  return *cache;
+}
+}  // namespace
+
+aqlprofile_agent_handle_t RegisterAgent(const aqlprofile_agent_info_v1_t* agent_info) {
+  aqlprofile_agent_handle_t agent_id;
+  AgentInfo int_agent_info;
+  int_agent_info.cu_num = agent_info->cu_num;
+  int_agent_info.se_num = agent_info->se_num;
+  int_agent_info.xcc_num = agent_info->xcc_num;
+  int_agent_info.shader_arrays_per_se = agent_info->shader_arrays_per_se;
+  int_agent_info.domain = agent_info->domain;
+  int_agent_info.bdf_id = agent_info->location_id;
+
+  auto len = strlen(agent_info->agent_gfxip);
+  memset(int_agent_info.gfxip, 0, sizeof(int_agent_info.gfxip));
+  memcpy(int_agent_info.gfxip, agent_info->agent_gfxip,
+         (len >= sizeof(int_agent_info.gfxip) ? sizeof(int_agent_info.gfxip) - 1 : len));
+  get_cache().add(agent_id.handle, int_agent_info);
+  return agent_id;
+}
+
+const AgentInfo* GetAgentInfo(aqlprofile_agent_handle_t agent_id) {
+  return get_cache().get(agent_id.handle);
+}
+
+}  // namespace aql_profile