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 1811–1819

streamingAttentionQuerySharedOrderedUncheckedSM86

Full file
1811def streamingAttentionQuerySharedOrderedUncheckedSM86 =
1812  (lambda unrestricted order : Nat . (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
1813    (lambda unrestricted keyValueHeads : Nat .
1814      (sqsPrologue order seq heads keyValueHeads
1815        (sm86ProgramAppend sqsBodyA
1816        (sm86ProgramAppend (sfFirstTail (sm86ProgramCount sqsBodyB))
1817        (sm86ProgramAppend sqsBodyB
1818        (sm86ProgramAppend (sfSecondTail (sm86ProgramCount sqsBodyA) (sm86ProgramCount sqsBodyB))
1819          sqsEpilogue)))))))))

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.