Normalize AMD Instinct accelerator names (#1817)

Closes: #1815
dstackai · Oct 15, 2024 · eebd572 · eebd572
1 parent f49a808
commit eebd572
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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"),
         ],
     )