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 1884–1894

streamingAttentionQuerySharedProjectionSM86

Full file
1884def streamingAttentionQuerySharedProjectionSM86 =
1885  (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
1886    (lambda unrestricted keyValueHeads : Nat . (lambda unrestricted width : Nat .
1887      (lambda erased admitted : (equal Nat (streamingAttentionQueryProjectionAdmitted seq heads keyValueHeads width) 1) .
1888        (sqsPrologue 1 seq heads keyValueHeads
1889          (sm86ProgramAppend sqsBodyA
1890          (sm86ProgramAppend (sfFirstTail (sm86ProgramCount sqsBodyB))
1891          (sm86ProgramAppend sqsBodyB
1892          (sm86ProgramAppend (sfSecondTail (sm86ProgramCount sqsBodyA) (sm86ProgramCount sqsBodyB))
1893            (sqProjectionEpilogue seq width
1894              (saOp (constructor SM86InstructionBody SM86Exit (saAfter saWaitAll)) sm86ProgramEmpty))))))))))))

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.