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 901–910

streamingAttentionForwardSharedOrderedWithSM86

Full file
901def streamingAttentionForwardSharedOrderedWithSM86 =
902  (lambda unrestricted rows : Nat . (lambda unrestricted order : Nat . (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
903    (lambda unrestricted keyValueHeads : Nat .
904      (let unrestricted bodyA = (sfBodyWithA rows) in (let unrestricted bodyB = (sfBodyWithB rows) in
905      (sfPrologueWith rows order seq heads keyValueHeads
906        (sm86ProgramAppend bodyA
907        (sm86ProgramAppend (sfFirstTail (sm86ProgramCount bodyB))
908        (sm86ProgramAppend bodyB
909        (sm86ProgramAppend (sfSecondTail (sm86ProgramCount bodyA) (sm86ProgramCount bodyB))
910          (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.