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 1500–1505

attentionHeadLayoutSM86RotatePlane

Full file
1500def attentionHeadLayoutSM86RotatePlane =
1501  (lambda unrestricted seq : Nat . (lambda unrestricted heads : Nat .
1502    (lambda unrestricted projectionWidth : Nat . (lambda unrestricted planeOffset : Nat .
1503      (lambda erased admitted :
1504        (equal Nat (attentionHeadLayoutSM86RotatePlaneAdmitted seq heads projectionWidth planeOffset) 1) .
1505        (attentionHeadLayoutSM86RotatePlaneUnchecked 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.