fixed runners and merge multi-file with new runners

gpu-mode · Jan 15, 2025 · f3d456d · f3d456d
1 parent 6a30fb7
commit f3d456d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git 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
@@ -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: