From ae4506d72251971005b123853951ca59e2ddfa87 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Tue, 2 Jun 2026 08:22:29 +0000 Subject: [PATCH] fix: w_gs is scalar not iterable --- tests/unit/test_fused_swiglu_kernel.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/unit/test_fused_swiglu_kernel.py b/tests/unit/test_fused_swiglu_kernel.py index 1e12c58f..b83ceacd 100644 --- a/tests/unit/test_fused_swiglu_kernel.py +++ b/tests/unit/test_fused_swiglu_kernel.py @@ -86,8 +86,7 @@ def test_fused_swiglu_compilation(): scale_b = assemble_scales_3d_side(w_sf) gsa = torch.full((num_experts,), x_gs, dtype=torch.float32, device=device) - gsb_vals = [float(g) for g in w_gs] # convert to Python floats - gsb = torch.tensor(gsb_vals, dtype=torch.float32, device=device) + gsb = torch.full((num_experts,), w_gs, dtype=torch.float32, device=device) # same gs for all experts # Pad activation x_padded = torch.zeros(128, K_packed, dtype=torch.uint8, device=device).view(torch.float4_e2m1fn_x2)