From 982f245c67f1729889be4a2e020f82aac975211e Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 06:47:15 +0000 Subject: [PATCH] fix: use correct Nvfp4Linear field names (fp4, scale_b, gsb) --- tests/unit/test_fused_router.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_fused_router.py b/tests/unit/test_fused_router.py index 4be3c1e3..135dd67c 100644 --- a/tests/unit/test_fused_router.py +++ b/tests/unit/test_fused_router.py @@ -38,9 +38,9 @@ def test_fused_router(): # The Nvfp4Linear expects stacked/contiguous weight tensors from dsv4.layers.linear import Nvfp4Linear gate_lin = Nvfp4Linear(in_features=K, out_features=N, device=device) - gate_lin._weight_fp4 = [w_fp4.contiguous()] - gate_lin._scale_b = [w_sf.contiguous()] - gate_lin._gsb = [ws2_val] + gate_lin.fp4 = [w_fp4.contiguous()] + gate_lin.scale_b = [w_sf.contiguous()] + gate_lin.gsb = [ws2_val] gate_lin._activation_global_scale = None # set at runtime gate_lin._ensure_stacked = lambda *a, **kw: None gate_lin._ensure_initialized = lambda *a, **kw: None