From 2abb4a19d91c7e3adc9238a08c417f49f23c85e0 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Mon, 1 Jun 2026 06:49:43 +0000
Subject: [PATCH] fix: set gs and ws2 fields for Nvfp4Linear in fused router
 test

---
 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 3250cef4..4fc3bd5f 100644
--- a/tests/unit/test_fused_router.py
+++ b/tests/unit/test_fused_router.py
@@ -40,10 +40,10 @@ def test_fused_router():
     gate_lin = Nvfp4Linear(in_features=K, out_features=N, device=device)
     gate_lin.fp4 = [w_fp4.contiguous()]
     gate_lin.sf = [w_sf.contiguous()]
-    gate_lin.gsb = [ws2_val]
+    gate_lin.gsb = None
+    gate_lin.gs = [1.0]  # default global scale
+    gate_lin.ws2 = [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
     gate_lin.finalize_weights()
 
     # e_bias