multiply by scheduling domain cpu count

1 files changed, 23 insertions, 1 deletions

diff --git a/distill_schedcat_profiles.py b/distill_schedcat_profiles.py
index fa3c482..8d34904 100755
--- a/distill_schedcat_profiles.py
+++ b/distill_schedcat_profiles.py
@@ -22,6 +22,23 @@ defaults = {
     'tag'    : '',
 }
 
+def ludwig_cpu_count(sched):
+    if 'L3' in sched:
+        cpus = 6
+    elif 'L2' in sched:
+        cpus = 2
+    elif 'P-'  in sched:
+        cpus= 1
+    else:
+        # global
+        cpus = 24
+    if '-RM' in sched and cpus == 24:
+        cpus -=1
+    return cpus
+
+HOSTS = {
+    'ludwig' : ludwig_cpu_count,
+}
 
 class DataFile(object):
     def __init__(self, fname):
@@ -50,8 +67,13 @@ def write_profiles(files):
     sched  = files[0].sched
     num    = len(files[0].data)
 
+    if host in HOSTS:
+        cpus = HOSTS[host](sched)
+    else:
+        cpus = 1
+
     def row(i, idx):
-        return [int(files[0].data[i][NUM_INDEX])] + [f.data[i][idx] for f in files]
+        return [int(files[0].data[i][NUM_INDEX]) * cpus] + [f.data[i][idx] for f in files]
 
     avg  = [row(i, AVG_INDEX) for i in xrange(num)]
     wc   = [row(i, WC_INDEX) for i in xrange(num)]