Skip to content

Commit 0aac552

Browse files
committed
Add MI355 to GPU mapping
1 parent ffd2636 commit 0aac552

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mojo/extensions.bzl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,7 @@ _gpu_toolchains_tag = tag_class(
203203
"l4": "nvidia:89",
204204
"mi300x": "amdgpu:gfx942",
205205
"mi325": "amdgpu:gfx942",
206+
"mi355": "amdgpu:gfx950",
206207
"rtx5090": "nvidia:120a",
207208
"metal3": "metal:30",
208209
"metal4": "metal:40",
@@ -227,6 +228,7 @@ _gpu_toolchains_tag = tag_class(
227228
"NVIDIA GeForce RTX 3090": "",
228229
"MI300X": "mi300x",
229230
"MI325": "mi325",
231+
"MI355": "mi355",
230232
"Navi": "radeon",
231233
"AMD Radeon Graphics": "radeon",
232234
"Metal 3": "metal3",

0 commit comments

Comments
 (0)