clr_amperf.py

#!/usr/bin/env python3
#
# reading the APERF and MPERF register via the amperf sysfs provided
# by the coolrfreq driver
#
# This code requires the coolrfreq driver
#
# Contact: Kazutomo Yoshii <ky@anl.gov>
#

import re, os, sys, time
import numpy as np
from clr_nodeinfo import *

class amperf_reader :

    def __init__ (self):
        self.ct = cputopology()

        self.init = False
        if os.path.exists('/sys/devices/system/cpu/cpu0/amperf'):
            self.init = True
        else:
            print('Warning: the amperf sysfs is unavailable')

        self.samples = [ None, None ]
        self.sidx = 0

        self.sample()
        time.sleep(.1)
        self.sample()

    def read(self):
        if not self.init:
            return None
        ret = {}
        for p in sorted(self.ct.pkgcpus.keys()):
            perpkg = {}
            for cpuid in self.ct.pkgcpus[p]:
                fn = "/sys/devices/system/cpu/cpu%d/amperf" % cpuid
                with open(fn, 'r') as f:
                    t = f.readline().split()
                    vals = []
                    for v in t:
                        vals.append(int(v))

                    perpkg['c%d' % cpuid] = vals
            ret['p%d' % p] = perpkg
            ret['time'] = time.time()
        return ret

    def sample(self):
        if not self.init:
            return None

        self.samples[self.sidx] = self.read()

        if self.sidx == 0:
            self.sidx = 1
        else:
            self.sidx = 0

    def firstidx(self):
        return self.sidx

    def secondidx(self):
        if self.sidx == 0:
            return 1
        return 0

    def getdiff(self):
        if not self.init:
            return None

        d = {}
        f = self.samples[self.firstidx()]
        s = self.samples[self.secondidx()]
        
        for kp in list(f.keys()):
            if kp == 'time':
                d['time'] = s[kp] - f[kp] 
            else:
                tmp = {}
                for kc in list(f[kp].keys()):
                    vals = []
                    for i in range(0,2):
                        vals.append(s[kp][kc][i] - f[kp][kc][i])
                    tmp[kc] = vals
                d[kp] = tmp
        return d

    def getavgGHz(self, d):
        if not self.init:
            return None

        ret = {}
        dt = d['time']

        for kp in list(d.keys()):
            if kp != 'time':
                tmp = {}
                for kc in list(d[kp].keys()):
                    # calc avgMHz delta_aperf / delta_t
                    tmp[kc] = d[kp][kc][0] / dt * 1e-9
                ret[kp] = tmp

        return ret

    def getpkgstats(self, p):
        if not self.init:
            return None

        ret = {}

        for k in list(p.keys()):
            tmp = []
            for c in list(p[k].keys()):
                tmp.append(p[k][c])

            ret[k] = [np.mean(tmp), np.std(tmp), np.min(tmp), np.max(tmp)]
        return ret

    def sample_and_json(self):
        if not self.init:
            return ''

        self.sample()
        d = self.getdiff()
        f = self.getavgGHz(d)
        s = self.getpkgstats(f)

        cnt = 0
        buf = '{"sample":"freq",'
        for kp in list(f.keys()):
            if cnt > 0:
                buf += ','
            cnt += 1
            buf += '"%s":{' % kp
            buf += '"mean":%.3lf,"std":%.3lf' % (s[kp][0], s[kp][1])
            for kc in list(f[kp].keys()):
                buf += ',"%s":%.3lf' % (kc,f[kp][kc])
            buf += '}'
        buf += '}'
        return buf

if __name__ == '__main__':

    amp = amperf_reader()

    while True:
        print(amp.sample_and_json())

        time.sleep(.5)