From 840e7155bf5473fb2137dfd3d5b904aa3e9defd9 Mon Sep 17 00:00:00 2001 From: Alex Trotta Date: Fri, 10 Oct 2025 16:06:19 -0400 Subject: [PATCH] Add MI355 to GPU mapping --- mojo/extensions.bzl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mojo/extensions.bzl b/mojo/extensions.bzl index 5647a4a..775e560 100644 --- a/mojo/extensions.bzl +++ b/mojo/extensions.bzl @@ -193,6 +193,7 @@ _gpu_toolchains_tag = tag_class( "l4": "nvidia:89", "mi300x": "amdgpu:gfx942", "mi325": "amdgpu:gfx942", + "mi355": "amdgpu:gfx950", "rtx5090": "nvidia:120a", "metal3": "metal:30", "metal4": "metal:40", @@ -217,6 +218,7 @@ _gpu_toolchains_tag = tag_class( "NVIDIA GeForce RTX 3090": "", "MI300X": "mi300x", "MI325": "mi325", + "MI355": "mi355", "Navi": "radeon", "AMD Radeon Graphics": "radeon", "Metal 3": "metal3",