Source/Packages

Realization.Nvidia.SM86.AdamWHalfSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/AdamWHalfSM86.alpha

205 lines38 declarations10.6 KiBSHA-256 557f99a5eb69

def · lines 72–87

ahPrologue

Full file
---- the prologue: T, the constants, the scalars, the count ----
72def ahPrologue = (lambda unrestricted iterations : Nat . (lambda unrestricted tail : (family SM86Program) .
73  (saS2R 0 (constructor SM86SpecialRegister SM86ThreadIdX)
74  (saS2R 1 (constructor SM86SpecialRegister SM86CooperativeThreadArrayIdX)
75  (saMovImm 3 16
76  (saMovImm 4 8
77  (saMovImm ahCounter iterations
78  (saMovConst 18 (ahScalar 0)
79  (saMovConst 19 (ahScalar 1)
80  (saMovConst 20 (ahScalar 2)
81  (saMovConst 21 (ahScalar 3)
82  (saMovConst 22 (ahScalar 4)
83  (saMovConst 23 (ahScalar 5)
84  (saMovConst 24 (ahScalar 6)
85  (saMovConst 25 (ahScalar 7)
86  (saMovImmAfter 5 0 saWait5
87  (saImad 2 1 ahThreads 0 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.