File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change 11"""
2- (beta) Explicit horizontal fusion with foreach_map and torch.compile
3- ============================================================
2+ Explicit horizontal fusion with foreach_map and torch.compile
3+ ===============================================================
44
55**Author:** `Michael Lazos <https://github.com/mlazos>`_
66"""
1313# allows conversion of any pointwise op in ``torch`` to a horiztonally fused foreach
1414# variant. In this tutorial, we will demonstrate how to implement the Adam optimizer
1515# with ``foreach_map`` to generate a fully fused kernel.
16- #
1716#
1817# .. note::
1918#
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+ #
2127
2228#####################################################################
2329# Model Setup
You can’t perform that action at this time.
0 commit comments