Skip to content

Commit 0a5c390

Browse files
committed
docs(relax): §16 parity reference correction — gridded bwd is checked vs serial-autograd GOLD directly (not MLX-proto); strictly stronger, per Verify audit
1 parent 6f0c8c5 commit 0a5c390

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

docs/RELAX-GRAPH-VS-MEGATRON.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -808,8 +808,11 @@ warning (`lane == 0`, also present in the §15 forward). `cuda_port_edits_needed
808808

809809
### Parity — FAIL: 7/8 grads pass, dD = 1.40e-3 EXCEEDS the 1e-3 gate (deterministic over 2 runs)
810810

811-
Per-grad max|abs| over ALL elements vs the MLX-proto-backward reference (the proto is 1.30e-4 vs
812-
the serial autograd GOLD), gate < 1e-3:
811+
Per-grad max|abs| over ALL elements vs the serial-autograd GOLD oracle **directly** — the probe
812+
takes the torch.autograd VJP of our exact serial recurrence (`_chunked_mamba3_diagonal_scan`) as the
813+
gold and compares the gridded B0/B1/B2 chain against it with no MLX-proto in the loop (strictly
814+
stronger than vs-proto: no proto↔gold slack). Gate < 1e-3, checked over every element (no row/head
815+
subset; per-grad NaN/Inf RAISES — the discipline the §13 forward inf-bug lacked):
813816

814817
| grad | max\|abs\| (run 1 / run 2) | gate | verdict |
815818
|---|---:|---:|:--|

0 commit comments

Comments
 (0)