Pointer analysis and store forwarding (remove loads after writes)

dag_in_context/src/ast.rs

@@ -129,8 +129,8 @@ pub fn not(e: RcExpr) -> RcExpr {
     RcExpr::new(Expr::Uop(UnaryOp::Not, e))
 }
 
-pub fn alloc(id: i64, amount: RcExpr, state: RcExpr, value_ty: BaseType) -> RcExpr {
-    RcExpr::new(Expr::Alloc(id, amount, state, value_ty))
+pub fn alloc(id: i64, amount: RcExpr, state: RcExpr, pointer_ty: BaseType) -> RcExpr {
+    RcExpr::new(Expr::Alloc(id, amount, state, pointer_ty))
 }
 
 pub fn free(ptr: RcExpr, state: RcExpr) -> RcExpr {

dag_in_context/src/greedy_dag_extractor.rs

@@ -569,7 +569,60 @@ impl CostModel for DefaultCostModel {
             "DoWhile" => 100., // TODO: we could make this more accurate
             "If" | "Switch" => 50.,
             // Unreachable
-            "HasType" | "HasArgType" | "ContextOf" | "NoContext" | "ExpectType" => 0.,
+            "I"
+            | "Infinity"
+            | "NegInfinity"
+            | "AddIntOrInfinity"
+            | "MaxIntOrInfinity"
+            | "MinIntOrInfinity"
+            | "MkIntInterval"
+            | "IntersectIntInterval"
+            | "UnionIntInterval"
+            | "AddIntIntervalToAll"
+            | "AddIntInterval"
+            | "AddIntIntervalToPtrPointees"
+            | "Intersect-PtrPointees"
+            | "PointsAnywhere"
+            | "ConsIfNonEmpty"
+            | "Cons-List<i64+IntInterval>"
+            | "Nil-List<i64+IntInterval>"
+            | "Length-List<i64+IntInterval>"
+            | "Concat-List<i64+IntInterval>"
+            | "RevConcat-List<i64+IntInterval>"
+            | "Rev-List<i64+IntInterval>"
+            | "UnionHelper-List<i64+IntInterval>"
+            | "Union-List<i64+IntInterval>"
+            | "IntersectHelper-List<i64+IntInterval>"
+            | "Intersect-List<i64+IntInterval>"
+            | "Cons-List<PtrPointees>"
+            | "Nil-List<PtrPointees>"
+            | "Length-List<PtrPointees>"
+            | "Concat-List<PtrPointees>"
+            | "RevConcat-List<PtrPointees>"
+            | "Rev-List<PtrPointees>"
+            | "Union-PtrPointees"
+            | "Zip<Union-PtrPointees>"
+            | "Zip<Intersect-PtrPointees>"
+            | "UnionPointees"
+            | "IntersectPointees"
+            | "UnwrapTuplePointsTo"
+            | "UnwrapPtrPointsTo"
+            | "PointeesDropFirst"
+            | "PointsToCellsAtIter"
+            | "PtrPointsTo"
+            | "TuplePointsTo"
+            | "GetPointees"
+            | "TypeToPointees"
+            | "PointsTo"
+            | "BaseTypeToPtrPointees"
+            | "TypeListToList<PtrPointees>"
+            | "HasType"
+            | "HasArgType"
+            | "ContextOf"
+            | "NoContext"
+            | "CellHasValues"
+            | "PointsToExpr"
+            | "ExpectType" => 0.,
             "ExprIsPure" | "ListExprIsPure" | "BinaryOpIsPure" | "UnaryOpIsPure" => 0.,
             "BodyContainsExpr" | "ScopeContext" => 0.,
             "Region" | "Full" | "IntB" | "BoolB" => 0.,

dag_in_context/src/lib.rs

@@ -51,6 +51,8 @@ pub fn prologue() -> String {
         include_str!("utility/canonicalize.egg"),
         include_str!("interval_analysis.egg"),
         include_str!("optimizations/switch_rewrites.egg"),
+        &optimizations::memory::rules(),
+        include_str!("optimizations/memory.egg"),
         &optimizations::loop_invariant::rules().join("\n"),
         include_str!("optimizations/loop_simplify.egg"),
         include_str!("optimizations/loop_unroll.egg"),