Fix imports: all functions are in cutedsl.bridge, not separate modules
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user