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 930–941

skLoopTail

Full file
930def skLoopTail = (lambda unrestricted bodyInstructions : Nat .
931  (saAddImm saKeyOffset saKeyOffset (naturalSaturatingSubtract 4294967296 (naturalMultiply skSubTile (naturalMultiply saHeadWidth saHalfBytes)))
932  (saAddImm saValueOffset saValueOffset (naturalSaturatingSubtract 4294967296 (naturalMultiply skSubTile saHalfBytes))
933  (saAddImm skLogOffset skLogOffset (naturalSaturatingSubtract 4294967296 (naturalMultiply skSubTile 4))
934  (saAddImm saCount saCount 4294967295
935  (saGreater saP1 saCount 0
936  (saWhen saP1
937    (constructor SM86InstructionBody SM86Branch
938      (saU (naturalSaturatingSubtract 4294967296 (naturalMultiply 16 (naturalAdd bodyInstructions 6))))
939      (sm86Unsigned32 (byte 255) (byte 255) (byte 131) (byte 3))
940      saPlain)
941    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.