From 991dff986df86c73d509c61b2ab8ac4632c017bb Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 22 Apr 2025 10:07:38 -0700 Subject: [PATCH 1/3] Add a note that foreach feature is a prototype --- recipes_source/foreach_map.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/recipes_source/foreach_map.py b/recipes_source/foreach_map.py index b8bf0aa2836..520748290f9 100644 --- a/recipes_source/foreach_map.py +++ b/recipes_source/foreach_map.py @@ -1,5 +1,5 @@ """ -(beta) Explicit horizontal fusion with foreach_map and torch.compile +Explicit horizontal fusion with foreach_map and torch.compile ============================================================ **Author:** `Michael Lazos `_ @@ -13,11 +13,17 @@ # allows conversion of any pointwise op in ``torch`` to a horiztonally fused foreach # variant. In this tutorial, we will demonstrate how to implement the Adam optimizer # with ``foreach_map`` to generate a fully fused kernel. -# # # .. note:: # -# This tutorial requires PyTorch 2.7.0 or later. +# This recipe describes a prototype feature. Prototype features are typically +# at an early stage for feedback and testing and are subject to change. +# +# Prerequisites +# ------------- +# +# * PyTorch v2.7.0 or later. +# ##################################################################### # Model Setup From 20f0bcec5ad8ffb475b28e2a9c7ec9b36d75a5f0 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 22 Apr 2025 10:08:46 -0700 Subject: [PATCH 2/3] Update recipes_source/foreach_map.py --- recipes_source/foreach_map.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/foreach_map.py b/recipes_source/foreach_map.py index 520748290f9..5b61aeefc37 100644 --- a/recipes_source/foreach_map.py +++ b/recipes_source/foreach_map.py @@ -22,7 +22,7 @@ # Prerequisites # ------------- # -# * PyTorch v2.7.0 or later. +# * PyTorch v2.7.0 or later # ##################################################################### From c6694a6d542fd710bea28b1a2922b6937208d8e1 Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Tue, 22 Apr 2025 10:19:03 -0700 Subject: [PATCH 3/3] Update recipes_source/foreach_map.py --- recipes_source/foreach_map.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/foreach_map.py b/recipes_source/foreach_map.py index 5b61aeefc37..0225a77e279 100644 --- a/recipes_source/foreach_map.py +++ b/recipes_source/foreach_map.py @@ -1,6 +1,6 @@ """ Explicit horizontal fusion with foreach_map and torch.compile -============================================================ +=============================================================== **Author:** `Michael Lazos `_ """