add inlining after optimization

dag_in_context/src/lib.rs

@@ -293,8 +293,11 @@ pub fn optimize(
 
         // if we are inlining, save the program
         // TODO we inline on the first pass, but this should be configurable from the schedule
-        // TODO inline
-        let inline_program = if i == 1000 { Some(res.clone()) } else { None };
+        let inline_program = if schedule.contains("INLINE") {
+            Some(res.clone())
+        } else {
+            None
+        };
 
         for func in fns {
             log::info!("Running pass {} on {}", i, func);

dag_in_context/src/optimizations/rec_to_loop.egg

@@ -86,8 +86,7 @@
    (= always-runs-len (tuple-length always-runs))
    (= start-ty (TupleT start-ty-list))
    (HasType body func-ty))
-  ((panic "good")
-    (let loop-ty
+  ((let loop-ty
      (TupleT (TLConcat start-ty-list (TCons (IntT) (TNil)))))
    ;; recursive case in the loop
    (let new-rec-case

dag_in_context/src/schedule.rs

@@ -50,7 +50,6 @@ fn optimizations() -> Vec<String> {
         "loop-inv-motion",
         "loop-strength-reduction",
         "loop-peel",
-        "rec-to-loop",
     ]
     .iter()
     .map(|opt| opt.to_string())
@@ -118,12 +117,22 @@ pub fn mk_sequential_schedule() -> Vec<String> {
 pub fn parallel_schedule() -> Vec<String> {
     let helpers = helpers();
 
-    vec![format!(
-        "
+    vec![
+        format!(
+            "
 (run-schedule
+    (repeat 2
+      {helpers}
+      swap-if)
     {helpers}
-    (repeat 2 swap-if)
-
+    rec-to-loop
+    {helpers})"
+        ),
+        format!(
+            "
+;; HACK: when INLINE appears in this string
+;; we perform inlining in this pass
+(run-schedule
     (repeat 2
         {helpers}
         all-optimizations
@@ -137,5 +146,6 @@ pub fn parallel_schedule() -> Vec<String> {
     {helpers}
 )
 "
-    )]
+        ),
+    ]
 }