Skip to content

fix(kernel): avoid nan in triton sampling softmax when a chunk starts with -inf - #654

Merged
jason-fxz merged 1 commit into
FlashML-org:mainfrom
ischencheng:fix/triton-softmax-neg-inf
Oct 10, 2026
Merged

jason-fxz merged 1 commit into
FlashML-org:mainfrom
ischencheng:fix/triton-softmax-neg-inf

Commits

  1. Commits on Oct 9, 2026