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 1628–1633

attentionHeadLayoutSM86TransposeValue

Full file
1628def attentionHeadLayoutSM86TransposeValue =
1629  (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
1630    (lambda unrestricted projectionWidth : Nat . (lambda unrestricted planeOffset : Nat .
1631      (lambda erased admitted :
1632        (equal Nat (attentionHeadLayoutSM86TransposeValueAdmitted seq heads projectionWidth planeOffset) 1) .
1633        (attentionHeadLayoutSM86TransposeValueUnchecked seq projectionWidth planeOffset))))))

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.