Skip softmax calibration via Triton kernel#1597

Merged

rohansjoshi merged 6 commits into

mainfrom

rohjoshi/triton-ss-calib

Jun 9, 2026

Commits on Jun 8, 2026

First commit
rohansjoshi
committed
Added decode calibration
rohansjoshi
committed
Fix decode calibration: full-cache kv_bound + 128x128 block to match PyTorch

kaix-nv
authored and
rohansjoshi
committed
Fix decode calibration: padded row in decode

kaix-nv
authored and
rohansjoshi
committed
Apply per-phase calibrated skip threshold at HF inference

kaix-nv
authored and
rohansjoshi
committed
Add sink-pattern decode calibration test (full cache + nonzero sparsity)

kaix-nv
authored and
rohansjoshi
committed