github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 13:58:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

nv: return real struct in _alloc_boot_struct (#11130)

Browse Source
This commit is contained in:
nimlgen
2025-07-08 20:04:43 +03:00
committed by GitHub
parent 3dfc0ff887
commit b516fe71b4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/support/nv/nvdev.py
View File

@@ -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}"