From 6656aa162ca0645834569d704536a54363387f37 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Fri, 4 Jul 2025 17:17:24 +0300 Subject: [PATCH] nv: enable huge pages (#11091) --- 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 80f8abdfb2..51e009f90c 100644 --- a/tinygrad/runtime/support/nv/nvdev.py +++ b/tinygrad/runtime/support/nv/nvdev.py @@ -52,7 +52,7 @@ class NVPageTableEntry: return (self.nvdev.NV_MMU_VER2_DUAL_PDE if self.lv == 3 else self.nvdev.NV_MMU_VER2_PDE).decode(self.entry(entry_id)) def is_huge_page(self, entry_id) -> bool: return (self.entry(entry_id) & 1 == 1) if self.lv <= 3 else True - def supports_huge_page(self, paddr:int): return self.lv >= 4 + def supports_huge_page(self, paddr:int): return self.lv >= 2 and paddr % self.nvdev.mm.pte_covers[self.lv] == 0 def valid(self, entry_id): if self.is_huge_page(entry_id): return self.read_fields(entry_id)['valid']