Skip to content

Commit 35c68ea

Browse files
authored
Add a note that foreach feature is a prototype (#3341)
* Add a note that foreach feature is a prototype
1 parent 71695c7 commit 35c68ea

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

recipes_source/foreach_map.py

+10-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
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
"""
@@ -13,11 +13,17 @@
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

0 commit comments

Comments
 (0)