fix: correct quantize function name in fused router test

2026-06-01 06:41:54 +00:00
parent 0873d65253
commit 27fd847dd0
1 changed files with 2 additions and 2 deletions
--- 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)