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 1651–1699

attentionHeadLayoutSM86InverseTransposeValueUnchecked

Full file
1651def attentionHeadLayoutSM86InverseTransposeValueUnchecked =
1652  (lambda unrestricted seq : Nat . (lambda unrestricted projectionWidth : Nat .
1653    (lambda unrestricted planeOffset : Nat .
1654      (attentionHeadLayoutSM86RotatePlaneSpecial (byte 0)
1655        (constructor SM86SpecialRegister SM86ThreadIdX)
1656      (attentionHeadLayoutSM86RotatePlaneSpecial (byte 1)
1657        (constructor SM86SpecialRegister SM86CooperativeThreadArrayIdX)
1658      (attentionHeadLayoutSM86RotatePlaneSpecial (byte 2)
1659        (constructor SM86SpecialRegister SM86CooperativeThreadArrayIdY)
1660      (attentionHeadLayoutSM86Next (attentionHeadLayoutSM86Move (byte 4) attentionHeadLayoutSM86U4)
1661      (attentionHeadLayoutSM86Next (attentionHeadLayoutSM86Move (byte 7) attentionHeadLayoutSM86U0)
1662      (attentionHeadLayoutSM86Next
1663        (constructor SM86InstructionBody SM86MoveImmediate
1664          (attentionHeadLayoutSM86R (byte 5)) attentionHeadLayoutSM86U0 attentionHeadLayoutSM86WaitSpecials)
1665      (attentionHeadLayoutSM86Next
1666        (attentionHeadLayoutSM86Move (byte 6) (attentionHeadLayoutSM86RotatePlaneWord planeOffset))
1667      (attentionHeadLayoutSM86Next
1668        (attentionHeadLayoutSM86IMAD (byte 3) (byte 1) attentionHeadLayoutSM86U2 (byte 7))
1669      (attentionHeadLayoutSM86Next
1670        (attentionHeadLayoutSM86IMAD (byte 16) (byte 0)
1671          (attentionHeadLayoutSM86RotatePlaneWord seq) (byte 3))
1672      (attentionHeadLayoutSM86Next
1673        (attentionHeadLayoutSM86IMAD (byte 16) (byte 2)
1674          (attentionHeadLayoutSM86RotatePlaneWord (naturalMultiply seq 64)) (byte 16))
1675      (attentionHeadLayoutSM86Next
1676        (attentionHeadLayoutSM86Wide (byte 10) (byte 16) (byte 4) attentionHeadLayoutSM86InputPointer)
1677      (attentionHeadLayoutSM86Next (attentionHeadLayoutSM86Load (byte 12) (byte 10) attentionHeadLayoutSM86U0)
1678      (attentionHeadLayoutSM86Next (attentionHeadLayoutSM86Load (byte 13) (byte 10) attentionHeadLayoutSM86U4)
1679      (attentionHeadLayoutSM86Next (attentionHeadLayoutSM86Move (byte 5) attentionHeadLayoutSM86U0)
1680      (attentionHeadLayoutSM86Next
1681        (constructor SM86InstructionBody SM86MoveImmediate
1682          (attentionHeadLayoutSM86R (byte 5)) attentionHeadLayoutSM86U0 attentionHeadLayoutSM86WaitLoads)
1683      (attentionHeadLayoutSM86Next
1684        (attentionHeadLayoutSM86IMAD (byte 8) (byte 3)
1685          (attentionHeadLayoutSM86RotatePlaneWord projectionWidth) (byte 6))
1686      (attentionHeadLayoutSM86Next
1687        (attentionHeadLayoutSM86IMAD (byte 8) (byte 2) attentionHeadLayoutSM86U64 (byte 8))
1688      (attentionHeadLayoutSM86Next
1689        (attentionHeadLayoutSM86IMAD (byte 8) (byte 0) attentionHeadLayoutSM86U1 (byte 8))
1690      (attentionHeadLayoutSM86Next
1691        (attentionHeadLayoutSM86Wide (byte 18) (byte 8) (byte 4) attentionHeadLayoutSM86FirstOutputPointer)
1692      (attentionHeadLayoutSM86Next
1693        (attentionHeadLayoutSM86Store (byte 18) (byte 12) attentionHeadLayoutSM86U0)
1694      (attentionHeadLayoutSM86Next
1695        (attentionHeadLayoutSM86Store (byte 18) (byte 13)
1696          (attentionHeadLayoutSM86RotatePlaneWord (naturalMultiply projectionWidth 4)))
1697      (attentionHeadLayoutSM86Next
1698        (constructor SM86InstructionBody SM86Exit sm86SafeControl)
1699        attentionHeadLayoutSM86End)))))))))))))))))))))))))

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.