AlphaGitHub
Source/Packages

Training.NativeObligation

packages/realizations/contracts/src/Training/NativeObligation.alpha

170 lines67 declarations8.5 KiBSHA-256 e731570b762e
Download exact fileView at revisionf5034773ad4bb52133

Complete file · line 6

NativeObligation.alpha

Definition view
1module Training.NativeObligation
2
3family NativeReadiness : Type 0
4constructor NativeEmitterAbsent
5constructor NativeEmitterPresent
6constructor NativeHardwareCorrect
7constructor NativePerformanceQualified
8
9end-family
10
11family NativeObligation : Type 0
12constructor NativeGradientAccumulatorClear
13constructor NativeIncomingGradientClear
14constructor NativeInputTransfer
15constructor NativeEmbeddingGather1024
16constructor NativeLayerNormForward1024
17constructor NativeLayerNormForward64
18constructor NativeLayerNormForward256
19constructor NativeDenseHMMAForward
20constructor NativeAttentionHMMAForward
21constructor NativeRoutedHMMAForward
22constructor NativeQKVRoPEForward
23constructor NativeCausalSoftmaxForward256
24constructor NativeHeadMajorTokenPermutation
25constructor NativeTokenHashRoute
26constructor NativeRoutingTelemetry
27constructor NativeRoutedGather
28constructor NativeGELUForward
29constructor NativeInverseRouteGather
30constructor NativeResidualAdd
31constructor NativeNegativeClassRNG
32constructor NativeClassRowGather256
33constructor NativePositiveClassDot
34constructor NativeImportanceCorrection
35constructor NativeCandidateConcatenate
36constructor NativeSampledCrossEntropy
37constructor NativeLossSeed
38constructor NativeCachedCrossEntropyBackward
39constructor NativeCandidateSplit
40constructor NativeAdjointCopy
41constructor NativeDenseHMMAReverse
42constructor NativeLayerNormBackward1024
43constructor NativeLayerNormBackward64
44constructor NativeLayerNormBackward256
45constructor NativeLayerNormParameterGradient1024
46constructor NativeLayerNormParameterGradient64
47constructor NativeLayerNormParameterGradient256
48constructor NativeQKVRoPEBackward
49constructor NativeAttentionHMMAReverse
50constructor NativeSoftmaxBackward256
51constructor NativeTokenHeadPermutationBackward
52constructor NativeRoutedGatherBackward
53constructor NativeRoutedHMMAReverse
54constructor NativeGELUBackward
55constructor NativeInverseRouteBackward
56constructor NativeResidualBackward
57constructor NativePositiveClassDotBackward
58constructor NativeClassRowScatter256
59constructor NativeAdjointAccumulate
60constructor NativeEmbeddingScatter1024
61constructor NativeGradientAccumulate
62constructor NativeReductionScratchClear
63constructor NativeGradientSquareReduction
64constructor NativeGradientPartialReduction
65constructor NativeDeviceClipScale
66constructor NativeDeviceGradientScale
67constructor NativeAdamW
68constructor NativeCrossEntropyRowsLaunch
69constructor NativeCrossEntropyPartialLaunch
70constructor NativeCrossEntropyFinalizeLaunch
71constructor NativeEmbeddingScatter512
72
73end-family
74
75def nativeObligationReadiness =
76  (lambda unrestricted obligation : (family NativeObligation) .
77    (eliminate
78      NativeObligation
79      (lambda unrestricted current : (family NativeObligation) . (family NativeReadiness))
80      obligation
81      (branch
82        NativeGradientAccumulatorClear
83        .
84        (constructor NativeReadiness NativePerformanceQualified))
85      (branch NativeIncomingGradientClear . (constructor NativeReadiness NativeEmitterPresent))
86      (branch NativeInputTransfer . (constructor NativeReadiness NativeEmitterPresent))
87      (branch NativeEmbeddingGather1024 . (constructor NativeReadiness NativePerformanceQualified))
88      (branch NativeLayerNormForward1024 . (constructor NativeReadiness NativePerformanceQualified))
89      (branch NativeLayerNormForward64 . (constructor NativeReadiness NativeHardwareCorrect))
90      (branch NativeLayerNormForward256 . (constructor NativeReadiness NativeHardwareCorrect))
91      (branch NativeDenseHMMAForward . (constructor NativeReadiness NativePerformanceQualified))
92      (branch NativeAttentionHMMAForward . (constructor NativeReadiness NativeEmitterPresent))
93      (branch NativeRoutedHMMAForward . (constructor NativeReadiness NativeEmitterPresent))
94      (branch NativeQKVRoPEForward . (constructor NativeReadiness NativeEmitterPresent))
95      (branch
96        NativeCausalSoftmaxForward256
97        .
98        (constructor NativeReadiness NativePerformanceQualified))
99      (branch NativeHeadMajorTokenPermutation . (constructor NativeReadiness NativeHardwareCorrect))
100      (branch NativeTokenHashRoute . (constructor NativeReadiness NativeEmitterPresent))
101      (branch NativeRoutingTelemetry . (constructor NativeReadiness NativeEmitterPresent))
102      (branch NativeRoutedGather . (constructor NativeReadiness NativeEmitterPresent))
103      (branch NativeGELUForward . (constructor NativeReadiness NativeHardwareCorrect))
104      (branch NativeInverseRouteGather . (constructor NativeReadiness NativeEmitterPresent))
105      (branch NativeResidualAdd . (constructor NativeReadiness NativeHardwareCorrect))
106      (branch NativeNegativeClassRNG . (constructor NativeReadiness NativeHardwareCorrect))
107      (branch NativeClassRowGather256 . (constructor NativeReadiness NativeHardwareCorrect))
108      (branch NativePositiveClassDot . (constructor NativeReadiness NativeHardwareCorrect))
109      (branch NativeImportanceCorrection . (constructor NativeReadiness NativeHardwareCorrect))
110      (branch NativeCandidateConcatenate . (constructor NativeReadiness NativeEmitterPresent))
111      (branch NativeSampledCrossEntropy . (constructor NativeReadiness NativeEmitterPresent))
112      (branch NativeLossSeed . (constructor NativeReadiness NativeEmitterPresent))
113      (branch NativeCachedCrossEntropyBackward . (constructor NativeReadiness NativeEmitterPresent))
114      (branch NativeCandidateSplit . (constructor NativeReadiness NativeEmitterPresent))
115      (branch NativeAdjointCopy . (constructor NativeReadiness NativeHardwareCorrect))
116      (branch NativeDenseHMMAReverse . (constructor NativeReadiness NativeEmitterPresent))
117      (branch
118        NativeLayerNormBackward1024
119        .
120        (constructor NativeReadiness NativePerformanceQualified))
121      (branch NativeLayerNormBackward64 . (constructor NativeReadiness NativeHardwareCorrect))
122      (branch NativeLayerNormBackward256 . (constructor NativeReadiness NativeHardwareCorrect))
123      (branch
124        NativeLayerNormParameterGradient1024
125        .
126        (constructor NativeReadiness NativePerformanceQualified))
127      (branch
128        NativeLayerNormParameterGradient64
129        .
130        (constructor NativeReadiness NativeHardwareCorrect))
131      (branch
132        NativeLayerNormParameterGradient256
133        .
134        (constructor NativeReadiness NativeHardwareCorrect))
135      (branch NativeQKVRoPEBackward . (constructor NativeReadiness NativeEmitterPresent))
136      (branch NativeAttentionHMMAReverse . (constructor NativeReadiness NativeHardwareCorrect))
137      (branch NativeSoftmaxBackward256 . (constructor NativeReadiness NativePerformanceQualified))
138      (branch
139        NativeTokenHeadPermutationBackward
140        .
141        (constructor NativeReadiness NativeHardwareCorrect))
142      (branch NativeRoutedGatherBackward . (constructor NativeReadiness NativeEmitterPresent))
143      (branch NativeRoutedHMMAReverse . (constructor NativeReadiness NativeEmitterPresent))
144      (branch NativeGELUBackward . (constructor NativeReadiness NativeHardwareCorrect))
145      (branch NativeInverseRouteBackward . (constructor NativeReadiness NativeEmitterPresent))
146      (branch NativeResidualBackward . (constructor NativeReadiness NativeHardwareCorrect))
147      (branch NativePositiveClassDotBackward . (constructor NativeReadiness NativeHardwareCorrect))
148      (branch NativeClassRowScatter256 . (constructor NativeReadiness NativeEmitterPresent))
149      (branch NativeAdjointAccumulate . (constructor NativeReadiness NativeHardwareCorrect))
150      (branch NativeEmbeddingScatter1024 . (constructor NativeReadiness NativeEmitterPresent))
151      (branch NativeGradientAccumulate . (constructor NativeReadiness NativePerformanceQualified))
152      (branch
153        NativeReductionScratchClear
154        .
155        (constructor NativeReadiness NativePerformanceQualified))
156      (branch
157        NativeGradientSquareReduction
158        .
159        (constructor NativeReadiness NativePerformanceQualified))
160      (branch
161        NativeGradientPartialReduction
162        .
163        (constructor NativeReadiness NativePerformanceQualified))
164      (branch NativeDeviceClipScale . (constructor NativeReadiness NativePerformanceQualified))
165      (branch NativeDeviceGradientScale . (constructor NativeReadiness NativePerformanceQualified))
166      (branch NativeAdamW . (constructor NativeReadiness NativePerformanceQualified))
167      (branch NativeCrossEntropyRowsLaunch . (constructor NativeReadiness NativeEmitterPresent))
168      (branch NativeCrossEntropyPartialLaunch . (constructor NativeReadiness NativeEmitterPresent))
169      (branch NativeCrossEntropyFinalizeLaunch . (constructor NativeReadiness NativeEmitterPresent))
170      (branch NativeEmbeddingScatter512 . (constructor NativeReadiness NativeEmitterPresent))))

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.