From 1a37b669223693e8bf891a4542c5c7d23e0b0ea3 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 15 May 2026 07:23:10 +0000 Subject: [PATCH] dang python --- src/nvfp4_megamoe_kernel/nvfp4_mega_moe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/nvfp4_megamoe_kernel/nvfp4_mega_moe.py b/src/nvfp4_megamoe_kernel/nvfp4_mega_moe.py index 971f33a2..7d2cc026 100644 --- a/src/nvfp4_megamoe_kernel/nvfp4_mega_moe.py +++ b/src/nvfp4_megamoe_kernel/nvfp4_mega_moe.py @@ -307,8 +307,8 @@ def nvfp4_mega_moe_full( l2_w, l2_sf = transformed_l2_weights # Expert sanity check — are experts actually distinct? - if not getattr(self, '_expert_sanity', False): - self._expert_sanity = True + if not getattr(nvfp4_mega_moe_full, '_expert_sanity', False): + nvfp4_mega_moe_full._expert_sanity = True for e in range(min(4, l1_w.shape[0])): w_sample = l1_w[e].view(torch.uint8)[:8, :8] sf_sample = l1_sf[e].to(torch.float32)[:4, :4]