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

def · lines 850–857

skLoadTileBase

Full file
A key launch may own a contiguous subset of key tiles. The caller derives the subset's base from its launch schedule; the ordinary unsplit kernel keeps using CTA x directly. This keeps all query-loop and output-address semantics in one implementation while bounding the work of one launch.
850def skLoadTileBase = (lambda unrestricted fromConstant : Nat .
851  (lambda unrestricted tail : (family SM86Program) .
852    (nat-eliminate (lambda unrestricted mode : Nat . (family SM86Program))
853      tail
854      (lambda unrestricted predecessor : Nat . (lambda unrestricted induction : (family SM86Program) .
855        (saMovConst saScratch (saArgument 13)
856          (saImad saTileIndex saScratch 1 saTileIndex tail))))
857      fromConstant)))

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.