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

def · lines 1005–1010

sfpValueSlice

Full file
V's slice kk (keys 16 kk.., columns 16 pair..) read transposed through sfT1, as sfLoadFragmentsTransposed's registers of that slice
1005def sfpValueSlice = (lambda unrestricted base : Nat . (lambda unrestricted kk : Nat .
1006  (lambda unrestricted tail : (family SM86Program) .
1007    (saFor 4 (lambda unrestricted pair : Nat .
1008      (saOp (sfpLdsm (sfpB kk (naturalMultiply 2 pair)) sfT1
1009        (naturalAdd base (naturalAdd (naturalMultiply kk (naturalMultiply 16 sfRowBytes)) (naturalMultiply pair 32))) 1 saWaitNone)))
1010      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.