File tree 1 file changed +10
-4
lines changed
1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change 1
1
"""
2
- (beta) Explicit horizontal fusion with foreach_map and torch.compile
3
- ============================================================
2
+ Explicit horizontal fusion with foreach_map and torch.compile
3
+ ===============================================================
4
4
5
5
**Author:** `Michael Lazos <https://github.com/mlazos>`_
6
6
"""
13
13
# allows conversion of any pointwise op in ``torch`` to a horiztonally fused foreach
14
14
# variant. In this tutorial, we will demonstrate how to implement the Adam optimizer
15
15
# with ``foreach_map`` to generate a fully fused kernel.
16
- #
17
16
#
18
17
# .. note::
19
18
#
20
- # This tutorial requires PyTorch 2.7.0 or later.
19
+ # This recipe describes a prototype feature. Prototype features are typically
20
+ # at an early stage for feedback and testing and are subject to change.
21
+ #
22
+ # Prerequisites
23
+ # -------------
24
+ #
25
+ # * PyTorch v2.7.0 or later
26
+ #
21
27
22
28
#####################################################################
23
29
# Model Setup
You can’t perform that action at this time.
0 commit comments