From 27fd847dd0dfa1129b4d7cbb419f40795c8f807d Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 06:41:54 +0000 Subject: [PATCH] fix: correct quantize function name in fused router test --- tests/unit/test_fused_router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_fused_router.py b/tests/unit/test_fused_router.py index 3c16a47b..4ffa557d 100644 --- a/tests/unit/test_fused_router.py +++ b/tests/unit/test_fused_router.py @@ -31,8 +31,8 @@ def test_fused_router(): # Create random BF16 gate weight and quantize to NVFP4 W_gate_bf16 = torch.randn(K, N, dtype=torch.bfloat16, device=device) - from dsv4.ops.quantize import quantize_weight_nvfp4 - w_fp4, w_sf, ws2_val, _ = quantize_weight_nvfp4(W_gate_bf16) + from dsv4.ops.quantize import quantize_weight_to_nvfp4 + w_fp4, w_sf, ws2_val, _ = quantize_weight_to_nvfp4(W_gate_bf16) # Build Nvfp4Linear for reference path gate_lin = Nvfp4Linear(in_features=K, out_features=N, device=device)