mlsanityData to 7 Oct 2026

P-Cast Precision in FP8 Attention: Sink-Induced Collapse and the Optimality of S=2^8

arXiv 2606.06521 ↗First posted 2 Jun 2026cs.AR cs.AI cs.DCcited by 0

Not on the map. Why?

Abstract (arXiv)

FP8 (E4M3) acceleration for attention computation offers significant throughput gains, but the 3-bit mantissa introduces precision challenges when the softmax probability matrix~P is cast to FP8 before the P · V matrix multiplication. We analyze two implementation choices that affect output precision under the Attention Sink phenomenon: (1)~the KV block iteration order, and (2) the static scaling factor applied to P before casting. We show that forward KV iteration causes P-collapse -- to leading order a fraction Phi(Delta + δk - 6.93 - ln S) of non-sink P values underflow to zero, where the small shift δk ≈ 1 (for ksink=4) is the expected within-sink-block score maximum -- and that reverse iteration removes it, with a zero-underflow guarantee when reverse is combined with S=256. We further give a constructive characterization of S = 256 = 28 as the static scale that simultaneously satisfies (i)~bit-exact IEEE 754 scaling, (ii) the lower envelope of a sawtooth function dp(S) over the E4M3 number line (dp = 2-4, the minimum worst-case quantization step), and (iii)~the maximum normal-range coverage among bit-exact (2k) scales (a non-bit-exact scale such as 448 attains slightly higher coverage; sec.5). Both optimizations are already deployed in FlashAttention-3/4 on engineering grounds; our contribution is a quantitative account of why these choices are good and a closed-form threshold Deltac = 6.93 + ln S - δk for predicting kernel-level precision loss. Kernel-faithful experiments (Q, K, V in FP32 to isolate the P-cast effect) show 3-10× MSE improvement at moderate sink strengths, and paired tests confirm both fixes saturate to the same precision floor when combined -- which motivated updating the hpc-ops kernel from S=1 to S=256.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 9 of its 18 references that are in the map. First posted on arXiv 2 Jun 2026.

Builds on

9 of 18 references.

References · 9

  1. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  2. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  3. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  4. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
  5. 2024Massive Activations in Large Language Modelscited by 277
  6. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
  7. 2022FP8 Formats for Deep Learningcited by 187
  8. 2025Why do LLMs attend to the first token?cited by 122
  9. 2024SageAttention2: Efficient Attention with Thorough Outlier Smoothing and Per-thread INT4 Quantizationcited by 113

Built on by

Not cited yet.