From b516fe71b479fcfe9559aca73ea082f02eca34c4 Mon Sep 17 00:00:00 2001
From: nimlgen <138685161+nimlgen@users.noreply.github.com>
Date: Tue, 8 Jul 2025 20:04:43 +0300
Subject: [PATCH] nv: return real struct in _alloc_boot_struct (#11130)

---
 tinygrad/runtime/support/nv/nvdev.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/runtime/support/nv/nvdev.py b/tinygrad/runtime/support/nv/nvdev.py
index 6f9d95ed7d..cd28c268f6 100644
--- a/tinygrad/runtime/support/nv/nvdev.py
+++ b/tinygrad/runtime/support/nv/nvdev.py
@@ -127,7 +127,7 @@ class NVDev(PCIDevImplBase):
   def _alloc_boot_struct(self, struct):
     va, paddrs = System.alloc_sysmem(sz:=ctypes.sizeof(type(struct)), contiguous=True)
     to_mv(va, sz)[:] = bytes(struct)
-    return struct, paddrs[0]
+    return type(struct).from_address(va), paddrs[0]
 
   def _download(self, file) -> str:
     url = f"https://raw.githubusercontent.com/NVIDIA/open-gpu-kernel-modules/e8113f665d936d9f30a6d508f3bacd1e148539be/{file}"