ampere_research/pytorch/power.sh
2024-12-11 16:36:06 -05:00

30 lines
801 B
Bash
Executable File

#! /bin/bash
baseline_time_s="$1"
arch=$(uname -m)
if [[ $arch = aarch64 ]]; then
iter=0
function aarch64_power {
sensors | awk '/CPU power:/ {print $3; exit}'
((iter++))
sleep 1s
}
baseline_time_s=-1
while [[ "$iter" -ne "$baseline_time_s" ]]; do
aarch64_power
done
elif [[ $arch = x86_64 ]]; then
if [[ "$baseline_time_s" -eq -1 ]]; then
#turbostat -s PkgWatt -i 1 2>/dev/null | awk -F: '/PkgWatt/ {getline; print $0}'
#turbostat -s PkgWatt -i 1 | awk '/PkgWatt/ {getline; print $0}'
turbostat -s PkgWatt ${@:2} | sed -n "/PkgWatt/{n;n;p}"
else
#turbostat -s PkgWatt -n "$baseline_time_s" -i 1 2>/dev/null | awk -F: '/PkgWatt/ {getline; print $0}'
turbostat -s PkgWatt -n "$baseline_time_s" -i 1 | sed -n "/PkgWatt/{n;n;p}"
fi
else
echo "Unrecognized arch!"
exit 1
fi