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

def · lines 468–519

attentionHeadLayoutSM86Prologue

Full file
shared prologue: R1 ← c[0x28]; R0 = tid; R1 = block; R4 = 4; R7 = 0; R9 = 2; R5 = mask; R2 = tid >> shift; R3 = tid & mask
468def attentionHeadLayoutSM86Prologue =
469  (lambda unrestricted shift : Byte .
470    (lambda unrestricted mask : (family SM86Unsigned32) .
471      (lambda unrestricted tail : (family SM86Program) .
472        (attentionHeadLayoutSM86Next
473          (constructor
474            SM86InstructionBody
475            SM86MoveConstant
476            (attentionHeadLayoutSM86R (byte 1))
477            (byte 0)
478            attentionHeadLayoutSM86ConstantBase
479            sm86SafeControl)
480          (attentionHeadLayoutSM86Next
481            (constructor
482              SM86InstructionBody
483              SM86SpecialToRegister
484              (attentionHeadLayoutSM86R (byte 0))
485              (constructor SM86SpecialRegister SM86ThreadIdX)
486              attentionHeadLayoutSM86SetSpecials)
487            (attentionHeadLayoutSM86Next
488              (constructor
489                SM86InstructionBody
490                SM86SpecialToRegister
491                (attentionHeadLayoutSM86R (byte 1))
492                (constructor SM86SpecialRegister SM86CooperativeThreadArrayIdX)
493                attentionHeadLayoutSM86SetSpecials)
494              (attentionHeadLayoutSM86Next
495                (attentionHeadLayoutSM86Move (byte 4) attentionHeadLayoutSM86U4)
496                (attentionHeadLayoutSM86Next
497                  (attentionHeadLayoutSM86Move (byte 7) attentionHeadLayoutSM86U0)
498                  (attentionHeadLayoutSM86Next
499                    (attentionHeadLayoutSM86Move (byte 5) mask)
500                    (attentionHeadLayoutSM86Next
501                      (attentionHeadLayoutSM86Move (byte 9) attentionHeadLayoutSM86U2)
502                      (attentionHeadLayoutSM86Next
503                        (constructor
504                          SM86InstructionBody
505                          SM86ShiftRightImmediate
506                          (attentionHeadLayoutSM86R (byte 2))
507                          (attentionHeadLayoutSM86R (byte 0))
508                          shift
509                          attentionHeadLayoutSM86WaitSpecials)
510                        (attentionHeadLayoutSM86Next
511                          (constructor
512                            SM86InstructionBody
513                            SM86LogicThreeInputTruthTable
514                            (attentionHeadLayoutSM86R (byte 3))
515                            (attentionHeadLayoutSM86R (byte 0))
516                            (attentionHeadLayoutSM86R (byte 5))
517                            (byte 192)
518                            sm86SafeControl)
519                          tail))))))))))))

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.