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 3035–3043

streamingAttentionKeySharedOrderedUncheckedSM86

Full file
3035def streamingAttentionKeySharedOrderedUncheckedSM86 =
3036  (lambda unrestricted order : Nat . (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
3037    (lambda unrestricted keyValueHeads : Nat . (lambda unrestricted fromConstant : Nat . (lambda unrestricted fromTileBase : Nat .
3038      (sksPrologue order seq heads keyValueHeads fromConstant fromTileBase
3039        (sm86ProgramAppend (sksBodyA seq)
3040        (sm86ProgramAppend (sfFirstTail (sm86ProgramCount (sksBodyB seq)))
3041        (sm86ProgramAppend (sksBodyB seq)
3042        (sm86ProgramAppend (sfSecondTail (sm86ProgramCount (sksBodyA seq)) (sm86ProgramCount (sksBodyB seq)))
3043          (sksEpilogue seq))))))))))))

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.