index working

dag_in_context/src/optimizations/conditional_invariant_code_motion.egg

@@ -1,4 +1,18 @@
 (ruleset cicm)
+(ruleset cicm-index)
+
+(relation InvCodeMotionCandidate (Expr Expr))
+
+(rule (
+        (= (TCPair t1 c1) (ExtractedExpr e1))
+        (= (TCPair t1 c2) (ExtractedExpr e2))
+        (ContextOf e1 (InIf true pred1 orig_ins3))
+        (ContextOf e2 (InIf true pred2 orig_ins4))
+        (!= e1 e2)
+     )
+     ((InvCodeMotionCandidate e1 e2))
+     :ruleset cicm-index)
+
 
 (rule (
         (= if_e (If pred orig_ins thn els))
@@ -67,25 +81,32 @@
       )
     :ruleset cicm)
 
+
+
 (rule (
+        (InvCodeMotionCandidate e1 e2)
         (= if_e (If pred orig_ins thn els))
         (HasArgType thn (TupleT tylist))
         (HasArgType els (TupleT tylist))
         (ContextOf if_e outer_ctx)
 
+        (ContextOf e1 (InIf true pred orig_ins))
+        (ContextOf e2 (InIf false pred orig_ins))
         (= e1 (Bop o x1 y1))
-        (HasType e1 (Base ty))
+
+        (= e2 (Bop o x2 y2))
+
         (= (TCPair t1 c1) (ExtractedExpr e1))
         (> 10 (Expr-size e1))
         (ExprIsPure e1)
-        (ContextOf e1 (InIf true pred orig_ins))
+        (HasType e1 (Base ty))
+
 
-        (= e2 (Bop o x2 y2))
+
         (HasType e2 (Base ty))
         (= (TCPair t2 c2) (ExtractedExpr e2))
         (> 10 (Expr-size e2))
         (ExprIsPure e2)
-        (ContextOf e2 (InIf false pred orig_ins))
 
         (= t1 t2)
       )

dag_in_context/src/schedule.rs

@@ -57,6 +57,10 @@ pub(crate) fn helpers() -> String {
      (saturate
        terms-helpers
        (saturate terms-helpers-helpers)))
+
+    ;; cicm index
+    cicm-index
+
     ;; memory-helpers TODO run memory helpers for memory optimizations
 
     ;; finally, subsume now that helpers are done