Add OutputDiscardCheckpoint#682

Open

AkshitaB wants to merge 19 commits into

mainfrom

akshitab/output-discard-checkpoint

Commits on May 15, 2026

Add OutputDiscardCheckpoint

authored and
AkshitaB
committed
Add Python-fallback test for OutputDiscardCheckpoint

AkshitaB
and
claude
committed
Document OutputDiscardCheckpoint usage and hook contract

AkshitaB
and
claude
committed

Commits on May 16, 2026

Fix OutputDiscardCheckpoint Python fallback for saved views

AkshitaB
and
claude
committed
Add ODC benchmark and FFN integration scripts

AkshitaB
and
claude
committed
Update changelog for OutputDiscardCheckpoint

AkshitaB
and
claude
committed
Mark OutputDiscardCheckpoint as @beta_feature

AkshitaB
and
claude
committed
[temporary] Run ODC benchmark and FFN integration check on GPU CI

AkshitaB
and
claude
committed
Expand ODC benchmark with block types and multi-layer stacks

AkshitaB
and
claude
committed
Add multi-layer to FFN integration check

AkshitaB
and
claude
committed
Group benchmark output by n_layers; add --iters

AkshitaB
and
claude
committed
increase layers and layers
AkshitaB
committed
Add fp32_softmax block and let --dtype sweep multiple precisions

AkshitaB
and
claude
committed
[temporary] Sweep bf16 and fp32 in ODC benchmark CI

AkshitaB
and
claude
committed
Fall through to baseline for Fp32CastBlock at fp32 base

AkshitaB
and
claude
committed