Init canonicalize infrastructure

.github/workflows/ci.yml

@@ -8,7 +8,7 @@ on:
   workflow_dispatch:
 
 env:
-  RUST_TOOLCHAIN: '1.80'
+  RUST_TOOLCHAIN: '1.81'
 
 jobs:
   test:

Cargo.toml

@@ -6,4 +6,7 @@ edition = "2021"
 license = "Apache-2.0 with LLVM-exceptions"
 
 [dependencies]
-anyhow = "1.0.86"
+anyhow = "1.0.86"
+
+[dev-dependencies]
+indoc = "2"

src/canonicalize.rs

@@ -0,0 +1,62 @@
+use crate::ir::ModuleOp;
+use crate::ir::Op;
+
+#[derive(Debug, PartialEq, Eq)]
+pub enum CanonicalizeResult {
+    Changed,
+    Unchanged,
+}
+
+pub trait Canonicalize {
+    fn canonicalize(&self, op: &dyn Op) -> CanonicalizeResult;
+}
+
+struct CanonicalizeOp;
+
+impl Canonicalize for CanonicalizeOp {
+    fn canonicalize(&self, op: &dyn Op) -> CanonicalizeResult {
+        op.canonicalize()
+    }
+}
+
+struct DeadCodeElimination;
+
+impl Canonicalize for DeadCodeElimination {
+    fn canonicalize(&self, _op: &dyn Op) -> CanonicalizeResult {
+        CanonicalizeResult::Unchanged
+    }
+}
+
+fn canonicalize_op(op: &dyn Op) -> CanonicalizeResult {
+    let canonicalizers: Vec<&dyn Canonicalize> = vec![&CanonicalizeOp]; // &DeadCodeElimination];
+    let mut changed = CanonicalizeResult::Unchanged;
+    let ops = op.ops();
+    for canonicalizer in &canonicalizers {
+        for nested_op in ops.iter() {
+            let nested_op = nested_op.read().unwrap();
+            let result = canonicalize_op(&*nested_op);
+            if result == CanonicalizeResult::Changed {
+                changed = result;
+            }
+            let result = canonicalizer.canonicalize(op);
+            if result == CanonicalizeResult::Changed {
+                changed = result;
+            }
+        }
+        let result = canonicalizer.canonicalize(op);
+        if result == CanonicalizeResult::Changed {
+            changed = result;
+        }
+    }
+    changed
+}
+
+pub fn canonicalize(op: &ModuleOp) {
+    let max_iterations = 16;
+    for _ in 0..max_iterations {
+        let result = canonicalize_op(op);
+        if result == CanonicalizeResult::Unchanged {
+            break;
+        }
+    }
+}