diff --git a/.github/workflows/runner.py b/.github/workflows/runner.py index c8f51c32..edba3c44 100644 --- a/.github/workflows/runner.py +++ b/.github/workflows/runner.py @@ -13,17 +13,21 @@ if config["lang"] == "cu": comp, run = run_cuda_script( - config.get("eval.cu", cu_eval), - config.get("reference.cuh", None), - config.get("submission.cuh", None), + { + "eval.cu": config.get("eval.cu", cu_eval), + "reference.cuh": config.get("reference.cuh", None), + "submission.cuh": config.get("submission.cuh", None), + }, arch=None, ) result = {"compile": asdict(comp), "run": asdict(run)} else: run = run_pytorch_script( - config.get("eval.py", py_eval), - config.get("reference.py", None), - config.get("submission.py", None), + { + "eval.py": config.get("eval.py", py_eval), + "reference.py": config.get("reference.py", None), + "submission.py": config.get("submission.py", None), + }, arch=None, ) result = {"run": asdict(run)} diff --git a/src/discord-cluster-manager/run_eval.py b/src/discord-cluster-manager/run_eval.py index cab78b66..9da8092c 100644 --- a/src/discord-cluster-manager/run_eval.py +++ b/src/discord-cluster-manager/run_eval.py @@ -220,7 +220,7 @@ def run_cuda_script( # # noqa: C901 run_result = run_program(["./eval.out"]) return compile_result, run_result - + # cleaning up all source files _before_ we let the user code run, just in # case there's something in there that the user isn't supposed to snoop finally: