fix: add quantize_to_nvfp4 import

2026-05-17 08:24:57 +00:00
parent 33e28100ee
commit 04999d86cf
1 changed files with 1 additions and 0 deletions
--- a/vllm/nvfp4_cutedsl.py
+++ b/vllm/nvfp4_cutedsl.py
@@ -18,6 +18,7 @@ import torch
 from cutedsl.bridge import (
    quantize_activation_nvfp4,
    quantize_weight_to_nvfp4,
+    quantize_to_nvfp4,
    make_b_k_major,
    assemble_scales_3d_side,
    run_nvfp4_grouped_gemm,