implements lp.distribute_loops

inducer · Jun 19, 2022 · 627ed8d · 627ed8d
1 parent 2c1a5f0
commit 627ed8d
Show file tree

Hide file tree

Showing 4 changed files with 442 additions and 0 deletions.
diff --git a/doc/ref_transform.rst b/doc/ref_transform.rst
@@ -78,6 +78,11 @@ Manipulating Instructions
 
 .. autofunction:: add_barrier
 
+Loop Distribution
+-----------------
+
+.. automodule:: loopy.transform.loop_distribution
+
 Registering Library Routines
 ----------------------------
 

diff --git a/loopy/__init__.py b/loopy/__init__.py
@@ -124,6 +124,8 @@
 from loopy.transform.pack_and_unpack_args import pack_and_unpack_args_for_call
 
 from loopy.transform.realize_reduction import realize_reduction
+from loopy.transform.loop_distribution import (distribute_loops,
+                                               IllegalLoopDistributionError)
 
 # }}}
 
@@ -253,6 +255,8 @@
 
         "pack_and_unpack_args_for_call",
 
+        "distribute_loops", "IllegalLoopDistributionError",
+
         # }}}
 
         "get_dot_dependency_graph",