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 1178–1186

streamingAttentionForwardPipelinedOrderedSM86

Full file
1178def streamingAttentionForwardPipelinedOrderedSM86 =
1179  (lambda unrestricted order : Nat . (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
1180    (lambda unrestricted keyValueHeads : Nat .
1181      (sfpPrologue order seq heads keyValueHeads
1182        (sm86ProgramAppend sfpBodyA
1183        (sm86ProgramAppend (sfFirstTail (sm86ProgramCount sfpBodyB))
1184        (sm86ProgramAppend sfpBodyB
1185        (sm86ProgramAppend (sfSecondTail (sm86ProgramCount sfpBodyA) (sm86ProgramCount sfpBodyB))
1186          (sfEpilogue seq heads))))))))))

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.