From f05a639da543b79bfc7d7ea0b3ca40f67e18ea53 Mon Sep 17 00:00:00 2001
From: Phil Carns <carns@mcs.anl.gov>
Date: Fri, 4 Oct 2024 10:33:54 -0400
Subject: [PATCH] basic power monitoring script

---
 src/quintain-cpu-power-watcher.sh | 61 +++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100755 src/quintain-cpu-power-watcher.sh

diff --git a/src/quintain-cpu-power-watcher.sh b/src/quintain-cpu-power-watcher.sh
new file mode 100755
index 0000000..57ce4ef
--- /dev/null
+++ b/src/quintain-cpu-power-watcher.sh
@@ -0,0 +1,61 @@
+#!/bin/bash
+
+# report power usage in watts over 1 second intervals for each powercap rapl
+# zone
+
+# derived from example at https://www.reddit.com/r/linuxquestions/comments/1dp81bq/hardware_power_consumption/`
+# changes:
+# - some bug fixes in the watts conversion
+# - track each powercap/rapl zone separately, and report its name
+# - use floating point timestamps and calculations
+
+# you can temporarily enable access to the required counters (i.e. on a
+# development laptop) by running the following:
+#
+# sudo chmod -R a+r /sys/class/powercap/intel-rapl
+
+declare -A zones
+declare -A old_zones
+
+old_time=0
+while :; do
+    # figure out how many zones are available; store current microjoules value
+    # for each in an associative array
+    for i in `ls /sys/class/powercap/intel-rapl/ |grep intel-rapl:`; do
+        name=`cat /sys/class/powercap/intel-rapl/$i/name`
+        energy_uj=`cat /sys/class/powercap/intel-rapl/$i/energy_uj`
+        zones[$name]=$energy_uj
+    done
+
+    # current timestamp in seconds and nanoseconds
+    time=$(date +%s.%N)
+
+    if [ ${#old_zones[@]} -ne 0 ]; then
+        # normal case.  calculate power in watts for each zone based on
+        # elapsed time and difference in reported microjoules
+        echo -n "$time"
+        for i in "${!zones[@]}"; do
+            watts=`echo "(${zones[$i]} - ${old_zones[$i]}) / (($time - $old_time) * 10^6)" | bc -l`
+            echo -e -n "\t$watts"
+        done
+        echo -e -n "\n"
+    else
+        # first loop iteration.  we don't have enough data yet to report
+        # watts, but we use this opportunity to print a header row
+        echo -n "# <timestamp>"
+        for i in "${!zones[@]}"; do
+            echo -e -n "\t<$i>"
+        done
+        echo -e -n "\n"
+    fi
+
+    # save old time and microjoules
+    old_time=$time
+    for i in "${!zones[@]}"; do
+        old_zones[$i]=${zones[$i]}
+    done
+
+    # one second interval
+    sleep 1
+done
+