Upload pipeline.py
Browse files- 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
|