refactor(hydroflow_plus): move rewrites to a submodule

hydroflow_plus/src/builder/built.rs

@@ -73,7 +73,7 @@ impl<'a> BuiltFlow<'a> {
     }
 
     pub fn with_default_optimize(self) -> BuiltFlow<'a> {
-        self.optimize_with(crate::persist_pullup::persist_pullup)
+        self.optimize_with(crate::rewrites::persist_pullup::persist_pullup)
     }
 
     fn into_deploy<D: LocalDeploy<'a>>(mut self) -> DeployFlow<'a, D> {

hydroflow_plus/src/lib.rs

@@ -34,10 +34,7 @@ pub use builder::FlowBuilder;
 
 pub mod ir;
 
-pub mod persist_pullup;
-pub mod profiler;
-
-pub mod properties;
+pub mod rewrites;
 
 mod staging_util;
 

hydroflow_plus/src/rewrites/mod.rs

@@ -0,0 +1,3 @@
+pub mod persist_pullup;
+pub mod profiler;
+pub mod properties;

hydroflow_plus/src/profiler.rs → hydroflow_plus/src/rewrites/profiler.rs

@@ -3,8 +3,8 @@ use std::cell::RefCell;
 use hydroflow::futures::channel::mpsc::UnboundedSender;
 use stageleft::*;
 
-use crate::ir::*;
-use crate::{profiler as myself, RuntimeContext};
+use crate::{ir::*, RuntimeContext};
+use super::profiler as myself; // TODO(shadaj): stageleft does not support `self::...`
 
 pub fn increment_counter(count: &mut u64) {
     *count += 1;