AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.StreamingAttentionSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/StreamingAttentionSM86.alpha

3,230 lines493 declarations193.3 KiBSHA-256 cc0366b3d857
Download exact fileView at revisionbcc11a4e7647d31628

def · lines 2721–2721

sksScoreGradientOffset

Full file
dS^T for the dQ product (streamingAttentionQueryFromScoresSM86): the [head][key][query] half plane's byte offset of the thread's key row and quad-lane column pair at the current sub-tile, and its address pair
2721def sksScoreGradientOffset : Nat = 239

The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.