PrIM -- first commit

1 files changed, 133 insertions, 0 deletions

diff --git a/run_strong_rank.py b/run_strong_rank.py
new file mode 100644
index 0000000..6763bc7
--- /dev/null
+++ b/run_strong_rank.py
@@ -0,0 +1,133 @@
+import os 
+import sys
+import getpass
+
+rootdir = "/" # Include path to repo
+print("Root dir: " + rootdir)
+
+applications = {"VA"       : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -i 2621440 -x 1"], 
+                "GEMV"     : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/gemv_host -m 8192 -n 1024"],
+                "SpMV"     : ["NR_DPUS=X NR_TASKLETS=Y make all", "./bin/host_code -v 0"],
+                "SEL"      : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -i 3932160 -x 1"],
+                "UNI"      : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -i 3932160 -x 1"],
+                "BS"       : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/bs_host -i 262144"],
+                "TS"       : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/ts_host -n 524288"],
+                "BFS"      : ["NR_DPUS=X NR_TASKLETS=Y make all", "./bin/host_code -v 0 -f data/loc-gowalla_edges.txt"],
+                "MLP"      : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/mlp_host -m 8192 -n 1024"],
+                "NW"       : ["NR_DPUS=X NR_TASKLETS=Y BL=Z BL_IN=2 make all", "./bin/nw_host -w 0 -e 1 -n 2560"],
+                "HST-S"    : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -b 256 -x 1"],
+                "HST-L"    : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -b 256 -x 1"],
+                "RED"      : ["NR_DPUS=X NR_TASKLETS=Y BL=Z VERSION=SINGLE make all", "./bin/host_code -w 0 -e 1 -i 6553600 -x 1"],
+                "SCAN-SSA" : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -i 3932160 -x 1"],
+                "SCAN-RSS" : ["NR_DPUS=X NR_TASKLETS=Y BL=Z make all", "./bin/host_code -w 0 -e 1 -i 3932160 -x 1"],
+                "TRNS"     : ["NR_DPUS=X NR_TASKLETS=Y make all", "./bin/host_code -w 0 -e 1 -p 64 -o 12288 -x 1"],}
+
+def run(app_name):
+    
+    NR_TASKLETS = [1, 2, 4, 8, 16]
+    NR_DPUS = [1, 4, 16, 64]
+    BL = [10] 
+
+    if app_name in applications:
+        print ("------------------------ Running: "+app_name+"----------------------")
+        print ("--------------------------------------------------------------------")
+        if(len(applications[app_name]) > 1):
+            make = applications[app_name][0]
+            run_cmd = applications[app_name][1]
+        
+            os.chdir(rootdir + "/"+app_name)
+            os.getcwd()
+        
+            os.system("make clean")
+
+            try:
+                os.mkdir(rootdir + "/"+ app_name +"/bin")
+            except OSError:
+                print ("Creation of the direction /bin failed")
+                
+            try:
+                os.mkdir(rootdir + "/"+ app_name +"/log")
+            except OSError:
+                print ("Creation of the direction /log failed")
+            
+            try:
+                os.mkdir(rootdir + "/"+ app_name +"/log/host")
+            except OSError: 
+                print ("Creation of the direction /log/host failed")
+
+            try:
+                os.mkdir(rootdir + "/"+ app_name +"/profile")
+            except OSError:
+                print ("Creation of the direction /profile failed")
+        
+
+            for r in NR_DPUS:
+                for t in NR_TASKLETS:
+                    for b in BL:
+                        m = make.replace("X", str(r))
+                        m = m.replace("Y", str(t))
+                        if (app_name == "NW"):
+                            if (r == 1):
+                                m = m.replace("Z", str(2560))
+                            elif (r == 4):
+                                m = m.replace("Z", str(640))
+                            elif (r == 16):
+                                m = m.replace("Z", str(160))
+                            elif (r == 64):
+                                m = m.replace("Z", str(40))
+                        else: 
+                            m = m.replace("Z", str(b))
+                        print ("Running = " + m) 
+                        try:
+                            os.system(m)
+                        except: 
+                            pass 
+
+                        r_cmd = run_cmd.replace("#ranks", str(r))
+                        r_cmd = r_cmd +  " >> profile/outs_tl"+str(t)+"_bl"+str(b)+"_dpus"+str(r) 
+                        
+                        print ("Running = " + app_name + " -> "+ r_cmd)
+                        try:
+                            os.system(r_cmd) 
+                        except:  
+                            pass 
+        else:
+            make = applications[app_name] 
+
+            os.chdir(rootdir + "/"+app_name)
+            os.getcwd()
+        
+            try:
+                os.mkdir(rootdir + "/"+ app_name +"/bin")
+                os.mkdir(rootdir + "/"+ app_name +"/log")
+                os.mkdir(rootdir + "/"+ app_name +"/log/host")
+                os.mkdir(rootdir + "/"+ app_name +"/profile")
+            except OSError:
+                print ("Creation of the direction failed")
+
+            print (make)    
+            os.system(make + ">& profile/out")
+
+    else:
+        print ( "Application "+app_name+" not available" )
+
+def main():
+    if(len(sys.argv) < 2):
+        print ("Usage: python run.py application")
+        print ("Applications available: ")
+        for key, value in applications.iteritems():
+            print (key )
+        print ("All")
+
+    else:
+        cmd = sys.argv[1]
+        print ("Application to run is: " + cmd )
+        if cmd == "All":
+            for key, value in applications.iteritems():
+                run(key)
+                os.chdir(rootdir)
+        else:
+            run(cmd)
+
+if __name__ == "__main__":
+    main()