From d53e9b27f74a3dc8f0d94c4413960c1d0212fb33 Mon Sep 17 00:00:00 2001
From: Dmitry Meyer <me@undef.im>
Date: Fri, 11 Oct 2024 08:36:12 +0000
Subject: [PATCH] Normalize AMD Instinct accelerator names

Closes: https://github.com/dstackai/dstack/issues/1815
---
 src/dstack/_internal/utils/gpu.py     | 1 +
 src/tests/_internal/utils/test_gpu.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/dstack/_internal/utils/gpu.py b/src/dstack/_internal/utils/gpu.py
index a9fc357123..938a84993a 100644
--- a/src/dstack/_internal/utils/gpu.py
+++ b/src/dstack/_internal/utils/gpu.py
@@ -31,6 +31,7 @@ def convert_nvidia_gpu_name(name: str) -> str:
 
 def convert_amd_gpu_name(name: str) -> str:
     """Convert asic.market_name from amd-smi to short version"""
+    name = name.replace("Instinct ", "")
     # https://github.com/ROCm/amdsmi/blob/52b3947/src/amd_smi/amd_smi_utils.cc#L558-L593
     if name == "MI300X-O":
         return "MI300X"
diff --git a/src/tests/_internal/utils/test_gpu.py b/src/tests/_internal/utils/test_gpu.py
index 1939f924af..e966e06274 100644
--- a/src/tests/_internal/utils/test_gpu.py
+++ b/src/tests/_internal/utils/test_gpu.py
@@ -27,6 +27,7 @@ def test_convert_nvidia_gpu_name(self, test_input, expected):
         ["test_input", "expected"],
         [
             ("MI300X-O", "MI300X"),
+            ("Instinct MI210", "MI210"),
             ("MI300A", "MI300A"),
         ],
     )