From a35b475d184c2fbe37bb33faa3763479c6974cc8 Mon Sep 17 00:00:00 2001 From: b1tg <33436708+b1tg@users.noreply.github.com> Date: Fri, 11 Apr 2025 00:33:02 +0800 Subject: [PATCH] fix am driver for gfx1201 (#9836) --- tinygrad/runtime/ops_amd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/ops_amd.py b/tinygrad/runtime/ops_amd.py index 412c5e9337..d491efba25 100644 --- a/tinygrad/runtime/ops_amd.py +++ b/tinygrad/runtime/ops_amd.py @@ -649,7 +649,7 @@ class KFDIface: class AMAllocationMeta: owner:AMDDevice; mapped_devs:list[AMDDevice]; mapping:AMMapping # noqa: E702 class PCIIface: - supported_devs:list[int] = [0x744c, 0x7480] + supported_devs:list[int] = [0x744c, 0x7480, 0x7550] vfio:bool = getenv("VFIO", 1) and HWInterface.exists("/dev/vfio/vfio") vfio_fd:HWInterface gpus:list[Any] = []