Upload 2 files

__main__.py

@@ -53,6 +53,12 @@ def _config_from_args(args) -> HPCConfig:
         cfg.therm_lambda_tilt = float(args.thermal_tilt)
     if args.pin_dc:
         cfg.pin_dc = True
+    if getattr(args, "segment", False):
+        cfg.segment = True
+    if getattr(args, "segment_k", None) is not None:
+        cfg.segment_k = int(args.segment_k)
+    if getattr(args, "segment_iters", None) is not None:
+        cfg.segment_iters = int(args.segment_iters)
     if args.no_q4_fallback:
         cfg.enable_q4_fallback = False
     if args.keep_embd:
@@ -304,6 +310,25 @@ def main() -> None:
         help="zero-DC pin (Q2_K free lunch): re-solve (d, m) on the exact dc=0 line "
         "after assembly/thermal; strictly lowers SSE while driving row-DC residual to ~0",
     )
+    parser.add_argument(
+        "--segment",
+        action="store_true",
+        help="band-major superblock segmentation for Q2_K / Q3_K (emits HPCQ.SEG.V1 label channels)",
+    )
+    parser.add_argument(
+        "--segment-k",
+        type=int,
+        default=None,
+        metavar="K",
+        help="band count k for superblock segmentation (default 16)",
+    )
+    parser.add_argument(
+        "--segment-iters",
+        type=int,
+        default=None,
+        metavar="N",
+        help="proxy fixed-point rounds for superblock segmentation (default 4)",
+    )
     parser.add_argument(
         "--format",
         choices=("mixed", "mixed_q1", "custom", "q1_0", "q3_k", "q4_0", "q2_k"),

gui.py

@@ -184,6 +184,9 @@ class HPCQuantizeGUI:
         self.thermal_var = tk.IntVar(value=0)
         self.thermal_tilt_var = tk.DoubleVar(value=1.0)
         self.pin_dc_var = tk.BooleanVar(value=False)
+        self.segment_var = tk.BooleanVar(value=False)
+        self.segment_k_var = tk.IntVar(value=16)
+        self.segment_iters_var = tk.IntVar(value=4)
 
         self.cli_preview_var = tk.StringVar()
         self.status_var = tk.StringVar(value="Ready.")
@@ -217,6 +220,11 @@ class HPCQuantizeGUI:
             self.promote_attn_q4_var,
             self.quant_format_var,
             self.thermal_var,
+            self.thermal_tilt_var,
+            self.pin_dc_var,
+            self.segment_var,
+            self.segment_k_var,
+            self.segment_iters_var,
         ):
             v.trace_add("write", lambda *_: self._update_cli_preview())
 
@@ -483,10 +491,23 @@ class HPCQuantizeGUI:
             text="Pin DC residual to 0 (Q2_K free lunch)",
             variable=self.pin_dc_var,
         ).grid(row=3, column=0, columnspan=2, sticky=tk.W, padx=4, pady=1)
+        ttk.Checkbutton(
+            spec_f,
+            text="Band-Major Segmentation (Q2_K / Q3_K)",
+            variable=self.segment_var,
+        ).grid(row=3, column=2, columnspan=2, sticky=tk.W, padx=4, pady=1)
+        ttk.Label(spec_f, text="Seg Bands (k):").grid(row=4, column=0, sticky=tk.W, padx=4, pady=2)
+        ttk.Spinbox(
+            spec_f, from_=2, to=16, increment=1, textvariable=self.segment_k_var, width=8
+        ).grid(row=4, column=1, sticky=tk.EW, padx=4, pady=2)
+        ttk.Label(spec_f, text="Seg Iters:").grid(row=4, column=2, sticky=tk.W, padx=4, pady=2)
+        ttk.Spinbox(
+            spec_f, from_=1, to=12, increment=1, textvariable=self.segment_iters_var, width=8
+        ).grid(row=4, column=3, sticky=tk.EW, padx=4, pady=2)
         ttk.Label(
             spec_f, text="(λ multiplier ≥1 during thermal anneal; ~1e2–1e3 ≈ zero-coherent-residual '0-error' anneal)",
             foreground="#93c5fd",
-        ).grid(row=4, column=0, columnspan=4, sticky=tk.W, padx=4, pady=(0, 2))
+        ).grid(row=5, column=0, columnspan=4, sticky=tk.W, padx=4, pady=(0, 2))
 
         # Right: GGUF Tensor Browser
         browser_frame = ttk.LabelFrame(right_pane, text="3. GGUF Tensor Inspector & Live Status", padding=8)
@@ -816,6 +837,9 @@ class HPCQuantizeGUI:
             therm_passes=max(0, int(self.thermal_var.get())),
             therm_lambda_tilt=max(1.0, float(self.thermal_tilt_var.get())),
             pin_dc=bool(self.pin_dc_var.get()),
+            segment=bool(self.segment_var.get()),
+            segment_k=max(2, int(self.segment_k_var.get())),
+            segment_iters=max(1, int(self.segment_iters_var.get())),
             quant_format=self.quant_format_var.get(),
             custom_quant_map=self._custom_map_for_plan() or {},
             enable_q4_fallback=bool(self.q4_fallback_var.get()) and self.quant_format_var.get() not in ("q2_k", "q3_k"),
@@ -869,6 +893,12 @@ class HPCQuantizeGUI:
             cmd += ["--thermal-tilt", str(float(self.thermal_tilt_var.get()))]
         if self.pin_dc_var.get():
             cmd += ["--pin-dc"]
+        if self.segment_var.get():
+            cmd += ["--segment"]
+            if int(self.segment_k_var.get()) != 16:
+                cmd += ["--segment-k", str(int(self.segment_k_var.get()))]
+            if int(self.segment_iters_var.get()) != 4:
+                cmd += ["--segment-iters", str(int(self.segment_iters_var.get()))]
         cmd.append("--verbose")
         return " ".join(shlex.quote(part) for part in cmd)