Feat: add mi300x8

src/libkernelbot/consts.py

@@ -20,6 +20,7 @@ class GitHubGPU(Enum):
     NVIDIA = "NVIDIA"
     MI300 = "MI300"
     MI250 = "MI250"
+    MI300x8 = "MI300x8"
 
 
 class ModalGPU(Enum):
@@ -118,6 +119,7 @@ class RankCriterion(Enum):
     "B200": "100",
     "NVIDIA": None,
     "MI300": None,
+    "MI300x8": None,
     "MI250": None,
 }
 

src/libkernelbot/launchers/github.py

@@ -53,11 +53,13 @@ async def run_submission(
         self, config: dict, gpu_type: GPU, status: RunProgressReporter
     ) -> FullResult:
         gpu_vendor = None
-        if gpu_type.value in ["MI300", "MI250"]:
+        if gpu_type.value in ["MI300", "MI250", "MI300x8"]:
             selected_workflow = "amd_workflow.yml"
-            runner_name = {"MI300": "amdgpu-mi300-x86-64", "MI250": "amdgpu-mi250-x86-64"}[
-                gpu_type.value
-            ]
+            runner_name = {
+                "MI300": "amdgpu-mi300-x86-64",
+                "MI250": "amdgpu-mi250-x86-64",
+                "MI300x8": "amdgpu-mi300-8-x86-64",
+            }[gpu_type.value]
             gpu_vendor = "AMD"
             requirements = AMD_REQUIREMENTS
         elif gpu_type.value == "NVIDIA":

src/libkernelbot/task.py

@@ -147,7 +147,8 @@ def make_task_definition(yaml_file: str | Path) -> LeaderboardDefinition:  # noq
         assert lang in ["CUDA", "Python", "Triton", "HIP"]
         templates[lang] = (root / source).read_text()
 
-    del raw["templates"]
+    if templates:
+        del raw["templates"]
     description = raw["description"]
     del raw["description"]
     task = LeaderboardTask.from_dict(raw)