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 1198–1208

attentionHeadLayoutSM86ExpectedInstructions

Full file
1198def attentionHeadLayoutSM86ExpectedInstructions =
1199  (lambda unrestricted kind : (family AttentionHeadLayoutSM86Kind) .
1200    (eliminate
1201      AttentionHeadLayoutSM86Kind
1202      (lambda unrestricted current : (family AttentionHeadLayoutSM86Kind) . Nat)
1203      kind
1204      (branch AttentionHeadLayoutQKRotary . attentionHeadLayoutSM86N63)
1205      (branch AttentionHeadLayoutValueTranspose . attentionHeadLayoutSM86N23)
1206      (branch AttentionHeadLayoutHeadToToken . attentionHeadLayoutSM86N23)
1207      (branch AttentionHeadLayoutTokenToHead . attentionHeadLayoutSM86N23)
1208      (branch AttentionHeadLayoutInverseRoPEQKVMerge . attentionHeadLayoutSM86N77)))

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.