Skip to content

Commit 840e715

Browse files
committed
Add MI355 to GPU mapping
1 parent a5fa81c commit 840e715

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
@@ -193,6 +193,7 @@ _gpu_toolchains_tag = tag_class(
193193
"l4": "nvidia:89",
194194
"mi300x": "amdgpu:gfx942",
195195
"mi325": "amdgpu:gfx942",
196+
"mi355": "amdgpu:gfx950",
196197
"rtx5090": "nvidia:120a",
197198
"metal3": "metal:30",
198199
"metal4": "metal:40",
@@ -217,6 +218,7 @@ _gpu_toolchains_tag = tag_class(
217218
"NVIDIA GeForce RTX 3090": "",
218219
"MI300X": "mi300x",
219220
"MI325": "mi325",
221+
"MI355": "mi355",
220222
"Navi": "radeon",
221223
"AMD Radeon Graphics": "radeon",
222224
"Metal 3": "metal3",

0 commit comments

Comments
 (0)