CompressedGemma commited on
Commit
4355fa3
·
verified ·
1 Parent(s): 59598e4

Upload pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +6 -0
pipeline.py CHANGED
@@ -887,6 +887,12 @@ class Q2KQuantizer:
887
  if cfg.segment:
888
  from .segmentation import apply_order, q2k_proxy_cfg, segment_stage
889
 
 
 
 
 
 
 
890
  X_p = x.reshape(n_blocks, QK_K)
891
  imp2d = (
892
  None
 
887
  if cfg.segment:
888
  from .segmentation import apply_order, q2k_proxy_cfg, segment_stage
889
 
890
+ _emit(
891
+ "Phase 0: Band-Major Segmentation & Lookup Table",
892
+ 0,
893
+ n_blocks,
894
+ f"Building {cfg.segment_table_size}-entry LUT & assigning {n_blocks:,} superblocks (k={cfg.segment_k})",
895
+ )
896
  X_p = x.reshape(n_blocks, QK_K)
897
  imp2d = (
898
  None