Indirect Einsum Backend

src/finchlite/algebra/__init__.py

@@ -4,6 +4,7 @@
     init_value,
     is_annihilator,
     is_associative,
+    is_commutative,
     is_distributive,
     is_idempotent,
     is_identity,
@@ -37,18 +38,16 @@
     "TensorFType",
     "TensorPlaceholder",
     "conjugate",
-    "conjugate",
-    "element_type",
     "element_type",
     "fill_value",
-    "fill_value",
     "first_arg",
     "fixpoint_type",
     "identity",
     "init_value",
     "is_annihilator",
     "is_associative",
     "is_distributive",
+    "is_commutative",
     "is_idempotent",
     "is_identity",
     "overwrite",

src/finchlite/algebra/algebra.py

@@ -345,6 +345,18 @@ def is_associative(op: Any) -> bool:
 register_property(np.logical_or, "__call__", "is_associative", lambda op: True)
 register_property(np.logical_xor, "__call__", "is_associative", lambda op: True)
 
+# Commutative properties
+
+for op in (operator.add, operator.mul, operator.and_, operator.or_, operator.xor,
+           np.logical_and, np.logical_or, np.logical_xor):
+    register_property(op, "__call__", "is_commutative", lambda op: True)
+
+for op in (operator.sub, operator.truediv, operator.floordiv, operator.pow,
+           operator.lshift, operator.rshift):
+    register_property(op, "__call__", "is_commutative", lambda op: False)
+
+def is_commutative(op: Any) -> bool:
+    return query_property(op, "__call__", "is_commutative")
 
 def is_identity(op: Any, val: Any) -> bool:
     """

src/finchlite/algebra/tensor.py

@@ -6,7 +6,6 @@
 from ..algebra import register_property
 from ..symbolic import FType, FTyped, ftype
 
-
 class TensorFType(FType, ABC):
     @property
     def ndim(self) -> np.intp:

src/finchlite/autoschedule/__init__.py

@@ -38,11 +38,21 @@
     push_fields,
     set_loop_order,
 )
+from .einsum import (
+    Einsum,
+    EinsumPlan,
+    EinsumCompiler,
+    EinsumScheduler
+)
 
 __all__ = [
     "Aggregate",
     "Alias",
     "DefaultLogicOptimizer",
+    "Einsum",
+    "EinsumPlan",
+    "EinsumCompiler",
+    "EinsumScheduler",
     "Field",
     "Literal",
     "LogicCompiler",
@@ -51,6 +61,7 @@
     "PostOrderDFS",
     "PostWalk",
     "PreWalk",
+    "PrintingLogicOptimizer",
     "Produces",
     "Query",
     "Reformat",