AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.StreamingAttentionSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/StreamingAttentionSM86.alpha

1,129 lines205 declarations66.1 KiBSHA-256 aefcbc3b00ca
Download exact fileView at revisionf5034773ad4bb52133

def · lines 609–619

sqLoopTail

Full file
609def sqLoopTail = (lambda unrestricted bodyInstructions : Nat .
610  (saAddImm saKeyOffset saKeyOffset (naturalMultiply saTile (naturalMultiply saHeadWidth saHalfBytes))
611  (saAddImm sqKTOffset sqKTOffset (naturalMultiply saTile saHalfBytes)
612  (saAddImm saCount saCount 4294967295
613  (saGreater saP1 saCount 0
614  (saWhen saP1
615    (constructor SM86InstructionBody SM86Branch
616      (saU (naturalSaturatingSubtract 4294967296 (naturalMultiply 16 (naturalAdd bodyInstructions 5))))
617      (sm86Unsigned32 (byte 255) (byte 255) (byte 131) (byte 3))
618      saPlain)
619    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.