Fix imports: all functions are in cutedsl.bridge, not separate modules

This commit is contained in:
2026-05-17 18:11:03 +00:00
parent 5e63a0d8a3
commit cd6144b832

View File

@@ -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