implements lp.distribute_loops

Author: kaushikcfd

doc/ref_transform.rst

@@ -78,6 +78,11 @@ Manipulating Instructions
 
 .. autofunction:: add_barrier
 
+Loop Distribution
+-----------------
+
+.. automodule:: loopy.transform.loop_distribution
+
 Registering Library Routines
 ----------------------------
 

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",