From a830d3788147f95fbc4d2a3b3331950eea506b92 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Mon, 14 Jul 2025 11:46:14 +0300 Subject: [PATCH] nv: check wpr2 is inited (#11226) --- tinygrad/runtime/support/nv/ip.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tinygrad/runtime/support/nv/ip.py b/tinygrad/runtime/support/nv/ip.py index 29bda7a0df..6b1253593d 100644 --- a/tinygrad/runtime/support/nv/ip.py +++ b/tinygrad/runtime/support/nv/ip.py @@ -158,6 +158,7 @@ class NV_FLCN(NV_IP): imemPa=self.desc_v3.IMEMPhysBase, imemVa=self.desc_v3.IMEMVirtBase, imemSz=self.desc_v3.IMEMLoadSize, dmemPa=self.desc_v3.DMEMPhysBase, dmemVa=0x0, dmemSz=self.desc_v3.DMEMLoadSize, pkc_off=self.desc_v3.PKCDataOffset, engid=self.desc_v3.EngineIdMask, ucodeid=self.desc_v3.UcodeId) + assert self.nvdev.NV_PFB_PRI_MMU_WPR2_ADDR_HI.read() != 0, "WPR2 is not initialized" self.reset(self.falcon, riscv=True)