@@ -143,6 +143,86 @@ Tensor saving and restoring functions
143143
144144.. autoapifunction :: transformer_engine.pytorch.restore_from_saved
145145
146+ Operation fuser
147+ ---------------
148+
149+ .. autoapiclass :: transformer_engine.pytorch.ops.Sequential
150+ :members: forward
151+
152+ .. autoapiclass :: transformer_engine.pytorch.ops.FusibleOperation
153+ :members: fuser_forward, fuser_backward
154+
155+ .. autoapiclass :: transformer_engine.pytorch.ops.BasicOperation
156+ :members: op_forward, op_backward
157+
158+ .. autoapiclass :: transformer_engine.pytorch.ops.FusedOperation
159+ :members: fuser_forward, fuser_backward
160+
161+ .. autoapifunction :: transformer_engine.pytorch.ops.register_forward_fusion
162+
163+ .. autoapifunction :: transformer_engine.pytorch.ops.register_backward_fusion
164+
165+ .. autoapiclass :: transformer_engine.pytorch.ops.Linear
166+
167+ .. autoapiclass :: transformer_engine.pytorch.ops.AddExtraInput
168+
169+ .. autoapiclass :: transformer_engine.pytorch.ops.AllGather
170+
171+ .. autoapiclass :: transformer_engine.pytorch.ops.AllReduce
172+
173+ .. autoapiclass :: transformer_engine.pytorch.ops.BasicLinear
174+ :members: _functional_forward, _functional_backward
175+
176+ .. autoapiclass :: transformer_engine.pytorch.ops.Bias
177+
178+ .. autoapiclass :: transformer_engine.pytorch.ops.ClampedSwiGLU
179+
180+ .. autoapiclass :: transformer_engine.pytorch.ops.ConstantScale
181+
182+ .. autoapiclass :: transformer_engine.pytorch.ops.Dropout
183+
184+ .. autoapiclass :: transformer_engine.pytorch.ops.GEGLU
185+
186+ .. autoapiclass :: transformer_engine.pytorch.ops.GELU
187+
188+ .. autoapiclass :: transformer_engine.pytorch.ops.GLU
189+
190+ .. autoapiclass :: transformer_engine.pytorch.ops.GroupedLinear
191+
192+ .. autoapiclass :: transformer_engine.pytorch.ops.Identity
193+
194+ .. autoapiclass :: transformer_engine.pytorch.ops.L2Normalization
195+
196+ .. autoapiclass :: transformer_engine.pytorch.ops.LayerNorm
197+
198+ .. autoapiclass :: transformer_engine.pytorch.ops.MakeExtraOutput
199+
200+ .. autoapiclass :: transformer_engine.pytorch.ops.QGELU
201+
202+ .. autoapiclass :: transformer_engine.pytorch.ops.QGEGLU
203+
204+ .. autoapiclass :: transformer_engine.pytorch.ops.Quantize
205+
206+ .. autoapiclass :: transformer_engine.pytorch.ops.ReGLU
207+
208+ .. autoapiclass :: transformer_engine.pytorch.ops.ReLU
209+
210+ .. autoapiclass :: transformer_engine.pytorch.ops.ReduceScatter
211+
212+ .. autoapiclass :: transformer_engine.pytorch.ops.Reshape
213+
214+ .. autoapiclass :: transformer_engine.pytorch.ops.RMSNorm
215+
216+ .. autoapiclass :: transformer_engine.pytorch.ops.SReGLU
217+
218+ .. autoapiclass :: transformer_engine.pytorch.ops.SReLU
219+
220+ .. autoapiclass :: transformer_engine.pytorch.ops.ScaledSwiGLU
221+
222+ .. autoapiclass :: transformer_engine.pytorch.ops.SiLU
223+
224+ .. autoapiclass :: transformer_engine.pytorch.ops.SwiGLU
225+
146226Deprecated functions
147227--------------------
148228
0 commit comments