AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.StreamingAttentionSM86

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

1,203 lines221 declarations70.9 KiBSHA-256 23d4e5e2aa2a
Download exact fileView at revision0dc6acfc09dcaa39d8

def · lines 360–370

saLoopTail

Full file
---- the loop's tail: the next key tile, one fewer left ----
360def saLoopTail = (lambda unrestricted seq : Nat . (lambda unrestricted bodyInstructions : Nat .
361  (saAddImm saKeyOffset saKeyOffset (naturalMultiply saTile (naturalMultiply saHeadWidth saHalfBytes))
362  (saAddImm saValueOffset saValueOffset (naturalMultiply saTile saHalfBytes)
363  (saAddImm saCount saCount 4294967295
364  (saGreater saP1 saCount 0
365  (saWhen saP1
366    (constructor SM86InstructionBody SM86Branch
367      (saU (naturalSaturatingSubtract 4294967296 (naturalMultiply 16 (naturalAdd bodyInstructions 5))))
368      (sm86Unsigned32 (byte 255) (byte 255) (byte 131) (byte 3))
369      saPlain)
370    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.