1 files changed, 21 insertions, 17 deletions

diff --git a/run_case_study_dag.py b/run_case_study_dag.py
index f098e9e..d9a0563 100755
--- a/run_case_study_dag.py
+++ b/run_case_study_dag.py
@@ -78,6 +78,7 @@ def main(pathName):
 
     task_launches = []
     max_core = 0
+    all_cores = set()
     # Load task specifications
     with open(pathName, "r") as file:
         for line in file:
@@ -86,6 +87,7 @@ def main(pathName):
             core_num = int(parameters[CORE_NUM]) % 16
             if max_core < core_num:
                 max_core = core_num
+            all_cores.add(core_num)
             # if parameters[0] == '1':
             #     other_core_num = int(parameters[OTHER_CORE_NUM]) % 16
             #     if other_core_num > max_core:
@@ -97,20 +99,22 @@ def main(pathName):
     ### Allocate Cache Ways ###
     run('mount -t resctrl resctrl /sys/fs/resctrl') 
 #    run('echo "L3:0=0000,1=0000,2=0000,3=0000" > /sys/fs/resctrl/schemata')
-    # for core in range(16):
-    #     run("mkdir -p /sys/fs/resctrl/core-{}".format(core))
-    #     ccx = core // 4
-    #     in_ccx = core % 4
-    #     mask = "0000"
-    #     mask = mask[:in_ccx] +'f'+ mask[in_ccx+1:]
-    #     if ccx == 0:
-    #         run('echo "L3:0=' + mask + ';1=0000;2=0000;3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
-    #     if ccx == 1:
-    #         run('echo "L3:0=0000;1='  + mask + ';2=0000;3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
-    #     if ccx == 2:
-    #         run('echo "L3:0=0000;1=0000;2='+ mask +';3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
-    #     if ccx == 3:
-    #         run('echo "L3:0=0000;1=0000;2=0000;3=' + mask + '" > /sys/fs/resctrl/core-{}/schemata'.format(core))
+    for core in range(16):
+        if core not in all_cores:
+            continue
+        run("mkdir -p /sys/fs/resctrl/core-{}".format(core))
+        ccx = core // 4
+        in_ccx = core % 4
+        mask = "0000"
+        mask = mask[:in_ccx] +'f'+ mask[in_ccx+1:]
+        if ccx == 0:
+            run('echo "L3:0=' + mask + ';1=0000;2=0000;3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
+        if ccx == 1:
+            run('echo "L3:0=0000;1='  + mask + ';2=0000;3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
+        if ccx == 2:
+            run('echo "L3:0=0000;1=0000;2='+ mask +';3=0000" > /sys/fs/resctrl/core-{}/schemata'.format(core))
+        if ccx == 3:
+            run('echo "L3:0=0000;1=0000;2=0000;3=' + mask + '" > /sys/fs/resctrl/core-{}/schemata'.format(core))
 
     # return 1
     run("rm -rf /dev/shm/*")
@@ -145,7 +149,7 @@ def main(pathName):
             if not pid_str:
                 print("Unable to launch {} as a solo task! Exiting...".format(task_name))
                 return 1
-            # run("echo " + pid_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
+            run("echo " + pid_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
         else:
             # Paried tasks
             other_binary_name = launch[OTHER_BINARY_NAME].strip()
@@ -171,8 +175,8 @@ def main(pathName):
             if not pid1_str or not pid2_str:
                 print("Unable to launch {} and {} in a pair! Exiting...".format(task_name, other_task_name))
                 return 1
-            # run("echo " + pid1_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
-            # run("echo " + pid2_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
+            run("echo " + pid1_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
+            run("echo " + pid2_str + " > /sys/fs/resctrl/core-" + str(phys_core) + "/tasks")
             pairID += 1
 
 if __name__ == "__main__":