From 1d308df9488a40744ca7295b4d809e07d9f51d30 Mon Sep 17 00:00:00 2001
From: Adam McDaniel <adam.mcdaniel17@gmail.com>
Date: Wed, 2 Jul 2025 09:18:23 -0400
Subject: [PATCH] Exposed the energy count and current socket power metrics to
 profilers like PAPI

---
 src/rocm_smi_device.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/rocm_smi_device.cc b/src/rocm_smi_device.cc
index 526600d505b..0afcea95c26 100755
--- a/src/rocm_smi_device.cc
+++ b/src/rocm_smi_device.cc
@@ -510,6 +510,9 @@ static const std::map<const char *, dev_depends_t> kDevFuncDependsMap = {
   {"rsmi_topo_numa_affinity_get",        {{kDevNumaNodeFName}, {}}},
   {"rsmi_dev_gpu_metrics_info_get",      {{kDevGpuMetricsFName}, {}}},
   {"rsmi_dev_gpu_reset",                 {{kDevGpuResetFName}, {}}},
+  {"rsmi_dev_energy_count_get",          {{kDevGpuMetricsFName}, {}}},
+  {"rsmi_dev_current_socket_power_get",  {{kDevGpuMetricsFName}, {}}},
+
   {"rsmi_dev_compute_partition_get",     {{kDevComputePartitionFName}, {}}},
   {"rsmi_dev_compute_partition_set",     {{kDevComputePartitionFName}, {}}},
   {"rsmi_dev_memory_partition_get",      {{kDevMemoryPartitionFName}, {}}},