Skip to content

Commit 1ef2ae0

Browse files
committed
Merge branch 'main' into tmoon/row-tp-layernorm-linear
2 parents 0e225c1 + e583222 commit 1ef2ae0

37 files changed

Lines changed: 1148 additions & 330 deletions

.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,4 +41,4 @@ compile_commands.json
4141
.nfs
4242
tensor_dumps/
4343
artifacts/
44-
*.DS_Store
44+
.DS_Store

3rdparty/cudnn-frontend

Submodule cudnn-frontend updated 185 files

build_tools/VERSION.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2.13.0.dev0
1+
2.14.0.dev0

docs/api/pytorch.rst

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
146226
Deprecated functions
147227
--------------------
148228

17.3 KB
Loading
28.3 KB
Loading

0 commit comments

Comments
 (0)