From cd6144b8329be1b7b6f1e42f46508ff82cc93f18 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 17 May 2026 18:11:03 +0000 Subject: [PATCH] Fix imports: all functions are in cutedsl.bridge, not separate modules --- tests/test_pipeline_real_weights.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py index 0fa05b11..244218e1 100644 --- a/tests/test_pipeline_real_weights.py +++ b/tests/test_pipeline_real_weights.py @@ -184,9 +184,10 @@ def main(): # ---- Reference: same runner but with dynamic gs (quantize_to_nvfp4) ---- print("\n--- Reference (dynamic gs via quantize_to_nvfp4) ---") # We'll use the same runner infrastructure but manually call the reference path - from cutedsl.quantize import quantize_to_nvfp4 - from cutedsl.gemm import run_nvfp4_grouped_gemm - from cutedsl.bridge import assemble_scales_3d_side, make_b_k_major, assemble_scales_3d_side + from cutedsl.bridge import ( + quantize_to_nvfp4, run_nvfp4_grouped_gemm, + assemble_scales_3d_side, make_b_k_major, + ) with torch.no_grad(): # Stack weights for GEMM