From 78028bb253d03891a11109d7b76aef1618cf08cb Mon Sep 17 00:00:00 2001
From: Xavier Routh <xrouth2@illinois.edu>
Date: Thu, 30 Jan 2025 10:01:03 -0600
Subject: [PATCH] remove loop canon from scheduler

---
 juno_scheduler/src/compile.rs |  3 ---
 juno_scheduler/src/pm.rs      | 27 ---------------------------
 2 files changed, 30 deletions(-)

diff --git a/juno_scheduler/src/compile.rs b/juno_scheduler/src/compile.rs
index 0b3264ac..7bf3c5c5 100644
--- a/juno_scheduler/src/compile.rs
+++ b/juno_scheduler/src/compile.rs
@@ -105,9 +105,6 @@ impl FromStr for Appliable {
             "forkify" => Ok(Appliable::Pass(ir::Pass::Forkify)),
             "gcm" | "bbs" => Ok(Appliable::Pass(ir::Pass::GCM)),
             "gvn" => Ok(Appliable::Pass(ir::Pass::GVN)),
-            "loop-canon" | "loop-canonicalization" => {
-                Ok(Appliable::Pass(ir::Pass::LoopCanonicalization))
-            }
             "infer-schedules" => Ok(Appliable::Pass(ir::Pass::InferSchedules)),
             "inline" => Ok(Appliable::Pass(ir::Pass::Inline)),
             "ip-sroa" | "interprocedural-sroa" => {
diff --git a/juno_scheduler/src/pm.rs b/juno_scheduler/src/pm.rs
index 33a7b480..76e81ee9 100644
--- a/juno_scheduler/src/pm.rs
+++ b/juno_scheduler/src/pm.rs
@@ -1785,33 +1785,6 @@ fn run_pass(
             // Put BasicBlocks back, since it's needed for Codegen.
             pm.bbs = bbs;
         }
-        Pass::LoopCanonicalization => {
-            assert!(args.is_empty());
-            pm.make_fork_join_maps();
-            pm.make_control_subgraphs();
-            pm.make_loops();
-            pm.make_typing();
-            let fork_join_maps = pm.fork_join_maps.take().unwrap();
-            let loops = pm.loops.take().unwrap();
-            let control_subgraphs = pm.control_subgraphs.take().unwrap();
-            let typing = pm.typing.take().unwrap();
-            for ((((func, fork_join_map), loop_nest), control_subgraph), typing) in
-                build_selection(pm, selection)
-                    .into_iter()
-                    .zip(fork_join_maps.iter())
-                    .zip(loops.iter())
-                    .zip(control_subgraphs.iter())
-                    .zip(typing.iter())
-            {
-                let Some(mut func) = func else {
-                    continue;
-                };
-                // changed |= loop_canonicalization(&mut func, control_subgraph, fork_join_map, loop_nest, typing);
-                // func.modified();
-            }
-            pm.delete_gravestones();
-            pm.clear_analyses();
-        }
     }
     println!("Ran Pass: {:?}", pass);
 
-- 
GitLab