84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
|
|
def performance_score(s):
|
|
|
|
s = str(s, "utf-8").splitlines()
|
|
for line in s:
|
|
if "Avg. throughput" in line:
|
|
l = line.split(' ')
|
|
for i in range(len(l)):
|
|
if l[i] == 'throughput:':
|
|
return float(l[i+1])
|
|
return 0
|
|
|
|
def validation_score(s):
|
|
s = str(s, "utf-8")
|
|
cnt = s.count("Result: VALID")
|
|
return float(cnt * 5)
|
|
|
|
# Avg. throughput: 4926.239883 GFLOPS
|
|
|
|
BASEDIR = "/home/kjp4155/submissions/final"
|
|
|
|
usernames = [x.split(',')[1] for x in open('accounts.csv').readlines()[1:]]
|
|
|
|
result = ""
|
|
|
|
# for user in usernames:
|
|
# submitpath1 = f"/home/{user}/submit/HW6/mat_mul.cpp"
|
|
# submitpath2 = f"/home/{user}/submit/HW6/kernel.cl"
|
|
|
|
# os.makedirs(f"{BASEDIR}/{user}", exist_ok=True)
|
|
# if os.path.exists(submitpath1):
|
|
# shutil.copyfile(submitpath1, f"{BASEDIR}/{user}/mat_mul.cpp")
|
|
# if os.path.exists(submitpath2):
|
|
# shutil.copyfile(submitpath2, f"{BASEDIR}/{user}/kernel.cl")
|
|
|
|
print(usernames)
|
|
|
|
|
|
|
|
for user in usernames:
|
|
|
|
if not os.path.isfile(f"{BASEDIR}/{user}/A/convolution.cpp"):
|
|
print(f"{user},0,0,0,0,-1")
|
|
result += f"{user},0,0,0,0,-1\n"
|
|
continue
|
|
|
|
# if not os.path.isfile(f"{BASEDIR}/{user}/A/convolution.cpp"):
|
|
# print(f"{user},0,0,0,0,-1")
|
|
# result += f"{user},0,0,0,0,-1\n"
|
|
# continue
|
|
|
|
# Prepare files
|
|
shutil.rmtree(f"{BASEDIR}/{user}/tmp-A", ignore_errors=True)
|
|
shutil.copytree(f"{BASEDIR}/skel", f"{BASEDIR}/{user}/tmp-A", dirs_exist_ok=True)
|
|
shutil.copytree(f"{BASEDIR}/{user}/A", f"{BASEDIR}/{user}/tmp-A", dirs_exist_ok=True)
|
|
subprocess.run(["make" ,"clean"], cwd=f"{BASEDIR}/{user}/tmp-A")
|
|
subprocess.run(["make" ,"-j"], cwd=f"{BASEDIR}/{user}/tmp-A")
|
|
sperfA = subprocess.run(["./run_performance_A.sh"], cwd=f"{BASEDIR}/{user}/tmp-A", capture_output=True)
|
|
svalA = subprocess.run(["./run_validate_A.sh"], cwd=f"{BASEDIR}/{user}/tmp-A", capture_output=True)
|
|
perf_scoreA = performance_score(sperfA.stdout)
|
|
val_scoreA = validation_score(svalA.stdout)
|
|
|
|
|
|
shutil.rmtree(f"{BASEDIR}/{user}/tmp-B", ignore_errors=True)
|
|
shutil.copytree(f"{BASEDIR}/skel", f"{BASEDIR}/{user}/tmp-B", dirs_exist_ok=True)
|
|
shutil.copytree(f"{BASEDIR}/{user}/B", f"{BASEDIR}/{user}/tmp-B", dirs_exist_ok=True)
|
|
subprocess.run(["make" ,"clean"], cwd=f"{BASEDIR}/{user}/tmp-B")
|
|
subprocess.run(["make" ,"-j"], cwd=f"{BASEDIR}/{user}/tmp-B")
|
|
sperfB = subprocess.run(["./run_performance_B.sh"], cwd=f"{BASEDIR}/{user}/tmp-B", capture_output=True)
|
|
svalB = subprocess.run(["./run_validate_B.sh"], cwd=f"{BASEDIR}/{user}/tmp-B", capture_output=True)
|
|
perf_scoreB = performance_score(sperfB.stdout)
|
|
val_scoreB = validation_score(svalB.stdout)
|
|
|
|
print(f"{user},{perf_scoreA},{val_scoreA},{perf_scoreB},{val_scoreB}")
|
|
result += f"{user},{perf_scoreA},{val_scoreA},{perf_scoreB},{val_scoreB}\n"
|
|
|
|
|
|
print(result)
|
|
|