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 978–983

sfpKeySlice

Full file
K's slice k (halves 16 k.., keys 16 pair..) as sfLoadFragments's registers of that slice, into its slice buffer
978def sfpKeySlice = (lambda unrestricted guard : Nat . (lambda unrestricted base : Nat . (lambda unrestricted k : Nat .
979  (lambda unrestricted tail : (family SM86Program) .
980    (saFor 4 (lambda unrestricted pair : Nat .
981      (sfpOp guard (sfpLdsm (sfpB k (naturalMultiply 2 pair)) sfLane
982        (naturalAdd base (naturalAdd (naturalMultiply pair (naturalMultiply 16 sfRowBytes)) (naturalMultiply k 32))) 0 saWaitNone)))
983      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.