Skip to content

Commit a992037

Browse files
authored
Feat: add mi300x8 (#346)
1 parent c787cfa commit a992037

3 files changed

Lines changed: 10 additions & 5 deletions

File tree

src/libkernelbot/consts.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ class GitHubGPU(Enum):
2020
NVIDIA = "NVIDIA"
2121
MI300 = "MI300"
2222
MI250 = "MI250"
23+
MI300x8 = "MI300x8"
2324

2425

2526
class ModalGPU(Enum):
@@ -118,6 +119,7 @@ class RankCriterion(Enum):
118119
"B200": "100",
119120
"NVIDIA": None,
120121
"MI300": None,
122+
"MI300x8": None,
121123
"MI250": None,
122124
}
123125

src/libkernelbot/launchers/github.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,13 @@ async def run_submission(
5353
self, config: dict, gpu_type: GPU, status: RunProgressReporter
5454
) -> FullResult:
5555
gpu_vendor = None
56-
if gpu_type.value in ["MI300", "MI250"]:
56+
if gpu_type.value in ["MI300", "MI250", "MI300x8"]:
5757
selected_workflow = "amd_workflow.yml"
58-
runner_name = {"MI300": "amdgpu-mi300-x86-64", "MI250": "amdgpu-mi250-x86-64"}[
59-
gpu_type.value
60-
]
58+
runner_name = {
59+
"MI300": "amdgpu-mi300-x86-64",
60+
"MI250": "amdgpu-mi250-x86-64",
61+
"MI300x8": "amdgpu-mi300-8-x86-64",
62+
}[gpu_type.value]
6163
gpu_vendor = "AMD"
6264
requirements = AMD_REQUIREMENTS
6365
elif gpu_type.value == "NVIDIA":

src/libkernelbot/task.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,8 @@ def make_task_definition(yaml_file: str | Path) -> LeaderboardDefinition: # noq
147147
assert lang in ["CUDA", "Python", "Triton", "HIP"]
148148
templates[lang] = (root / source).read_text()
149149

150-
del raw["templates"]
150+
if templates:
151+
del raw["templates"]
151152
description = raw["description"]
152153
del raw["description"]
153154
task = LeaderboardTask.from_dict(raw)

0 commit comments

Comments
 (0)