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 1150–1158

sfpKeyCopy

Full file
K's four chunk rows of its next tile alone
1150def sfpKeyCopy = (lambda unrestricted keyBuffer : Nat . (lambda unrestricted keyGuard : Nat .
1151  (lambda unrestricted tail : (family SM86Program) .
1152    (saFor 4 (lambda unrestricted i : Nat .
1153      (sfWide (sfPair i) (sfKeyOffset i) (saArgument 1) (naturalSelect (naturalIsZero i) saWait5 saWaitNone)))
1154    (saFor 4 (lambda unrestricted j : Nat .
1155      (sfpOp keyGuard
1156        (sfCopyBody (naturalAdd keyBuffer (naturalMultiply j (naturalMultiply 16 sfRowBytes)))
1157          j (naturalSelect (naturalIsZero j) saWaitNone saWait5))))
1158    (sfCommit tail))))))

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.