From 8de950d9ca404e65bbf9eaa602a362e6df941ba7 Mon Sep 17 00:00:00 2001 From: Donato Capitella Date: Mon, 9 Mar 2026 12:13:27 +0000 Subject: [PATCH] feat: Override `_get_gcn_arch` function to return "gfx1151" and rename the original implementation to `_old_get_gcn_arch`. --- scripts/patch_strix.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/patch_strix.py b/scripts/patch_strix.py index 94c8c33..056d23f 100644 --- a/scripts/patch_strix.py +++ b/scripts/patch_strix.py @@ -25,6 +25,7 @@ def patch_vllm(): txt = p_rocm.read_text() header = 'import sys\nfrom unittest.mock import MagicMock\nsys.modules["amdsmi"] = MagicMock()\n' txt = header + txt + txt = txt.replace('def _get_gcn_arch() -> str:', 'def _get_gcn_arch() -> str:\n return "gfx1151"\n\ndef _old_get_gcn_arch() -> str:') txt = re.sub(r'device_type = .*', 'device_type = "rocm"', txt) txt = re.sub(r'device_name = .*', 'device_name = "gfx1151"', txt) txt += '\n def get_device_name(self, device_id: int = 0) -> str:\n return "AMD-gfx1151"\n'