Skip to content

Commit 39b9020

Browse files
committed
add mi300x8
1 parent 776931b commit 39b9020

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/libkernelbot/consts.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ class SchedulerType(Enum):
1919
class GitHubGPU(Enum):
2020
NVIDIA = "NVIDIA"
2121
MI300 = "MI300"
22+
MI300x8 = "MI300x8"
2223
MI250 = "MI250"
2324

2425

src/libkernelbot/launchers/github.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,9 @@ 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"}[
58+
runner_name = {"MI300": "amdgpu-mi300-x86-64", "MI250": "amdgpu-mi250-x86-64", "MI300x8": "amdgpu-mi300-8-x86-64"}[
5959
gpu_type.value
6060
]
6161
gpu_vendor = "AMD"

0 commit comments

Comments
 (0)