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 1336–1346

sqLoopTail

Full file
1336def sqLoopTail = (lambda unrestricted bodyInstructions : Nat .
1337  (saAddImm saKeyOffset saKeyOffset (naturalMultiply saTile (naturalMultiply saHeadWidth saHalfBytes))
1338  (saAddImm sqKTOffset sqKTOffset (naturalMultiply saTile saHalfBytes)
1339  (saAddImm saCount saCount 4294967295
1340  (saGreater saP1 saCount 0
1341  (saWhen saP1
1342    (constructor SM86InstructionBody SM86Branch
1343      (saU (naturalSaturatingSubtract 4294967296 (naturalMultiply 16 (naturalAdd bodyInstructions 5))))
1344      (sm86Unsigned32 (byte 255) (byte 255) (byte 131) (byte 3))
1345      saPlain)
1346    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.