Split common into submodules

Author: adam-smnk

lighthouse/schedule/__init__.py

@@ -1,6 +1,6 @@
 from .builders import create_schedule
 from .builders import create_named_sequence
-from .common import hoist_loops
+from .hoisting import hoist_loops
 from .linalg import linalg_to_generic
 from .linalg import linalg_to_category
 from .linalg import linalg_to_named

lighthouse/schedule/common.py lighthouse/schedule/hoisting.pylighthouse/schedule/common.py renamed to lighthouse/schedule/hoisting.py

@@ -1,6 +1,6 @@
-from .common import cleanup
-from .common import loop_hoisting
-from .common import match_op
+from .cleanup import cleanup
+from .hoisting import loop_hoisting
+from .matchers import match_op
 from .linalg import linalg_morph_ops
 from .tiling import tile_ops
 from .packing import block_pack_matmuls

lighthouse/transform/__init__.py

@@ -0,0 +1,14 @@
+from mlir import ir
+from mlir.dialects import transform
+
+
+def cleanup(target):
+    """
+    Apply canonicalization patterns.
+
+    Args:
+        target: Handle to target
+    """
+    transform.apply_cse(target)
+    with ir.InsertionPoint(transform.ApplyPatternsOp(target).patterns):
+        transform.apply_patterns_canonicalization()

lighthouse/transform/cleanup.py

@@ -0,0 +1,13 @@
+from mlir.dialects import transform
+from mlir.dialects.transform import loop
+
+
+def loop_hoisting(target):
+    """
+    Apply loop hoisting.
+
+    Args:
+        target: Handle to target
+    """
+    transform.apply_licm(target)
+    loop.loop_hoist_loop_invariant_subsets(target)

lighthouse/transform/hoisting.py

@@ -1,6 +1,5 @@
 from mlir import ir
 from mlir.dialects import transform
-from mlir.dialects.transform import loop
 from mlir.dialects.transform import structured
 
 
@@ -26,26 +25,3 @@ def match_op(
     if isinstance(target_op, str):
         target_op = [target_op]
     return structured.MatchOp.match_op_names(target, target_op)
-
-
-def cleanup(target):
-    """
-    Apply canonicalization patterns.
-
-    Args:
-        target: Handle to target
-    """
-    transform.apply_cse(target)
-    with ir.InsertionPoint(transform.ApplyPatternsOp(target).patterns):
-        transform.apply_patterns_canonicalization()
-
-
-def loop_hoisting(target):
-    """
-    Apply loop hoisting.
-
-    Args:
-        target: Handle to target
-    """
-    transform.apply_licm(target)
-    loop.loop_hoist_loop_invariant_subsets(target)