implements loop fusion transformation

inducer · Dec 10, 2021 · 1a5bc7e · 1a5bc7e
1 parent c78f2c9
commit 1a5bc7e
Show file tree

Hide file tree

Showing 3 changed files with 661 additions and 0 deletions.
diff --git a/doc/ref_transform.rst b/doc/ref_transform.rst
@@ -141,4 +141,10 @@ TODO: Matching instruction tags
 
 .. automodule:: loopy.match
 
+
+Fusing Loops
+------------
+
+.. automodule:: loopy.transform.loop_fusion
+
 .. vim: tw=75:spell
diff --git a/loopy/__init__.py b/loopy/__init__.py
@@ -156,6 +156,9 @@
 
 from loopy.tools import Optional, t_unit_to_python
 
+from loopy.transform.loop_fusion import (get_kennedy_unweighted_fusion_candidates,
+                                         rename_inames_in_batch)
+
 
 __all__ = [
         "TaggedVariable", "Reduction", "LinearSubscript", "TypeCast",
@@ -248,6 +251,9 @@
 
         "pack_and_unpack_args_for_call",
 
+        "rename_inames_in_batch",
+        "get_kennedy_unweighted_fusion_candidates",
+
         # }}}
 
         "get_dot_dependency_graph",