AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.AttentionHeadLayoutSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/AttentionHeadLayoutSM86.alpha

1,710 lines162 declarations85.3 KiBSHA-256 ad8cef82929b
Download exact fileView at revisionf5034773ad4bb52133

def · lines 251–251

attentionHeadLayoutSM86FirstOutputArgument

Full file
Ordered launch arguments shared by the specialised Q, K and V layouts. V reads through the input pointer but has no table operands.
251def attentionHeadLayoutSM86FirstOutputArgument : Nat = 0

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.