Automated Filament Scheduling

Cargo.toml

@@ -31,10 +31,12 @@ smallvec = "1"
 topological-sort = "0.2"
 atty = "0.2"
 lazy_static = "1.4"
-easy-smt = { version = "0.2.1" }
+easy-smt = "0.2.3"
 struct-variant = "1.0"
+tempfile = "3.13.0"
 strum = "0.26.3"
 strum_macros = "0.26.3"
+ordered-float = "5.0.0"
 boxcar = "0.2.10"
 
 fil-utils = { version = "0.1.0", path = "crates/utils" }

crates/ast/src/signature.rs

@@ -182,6 +182,14 @@ impl Signature {
     pub fn outputs(&self) -> impl Iterator<Item = &Loc<PortDef>> {
         self.ports[self.outputs_idx..].iter()
     }
+    /// Inputs of this signature
+    pub fn inputs_mut(&mut self) -> impl Iterator<Item = &mut Loc<PortDef>> {
+        self.ports[..self.outputs_idx].iter_mut()
+    }
+    /// Outputs of this signature
+    pub fn outputs_mut(&mut self) -> impl Iterator<Item = &mut Loc<PortDef>> {
+        self.ports[self.outputs_idx..].iter_mut()
+    }
     /// Iterator over all the ports of this signature
     pub fn ports(&self) -> &Vec<Loc<PortDef>> {
         &self.ports

crates/filament/Cargo.toml

@@ -38,6 +38,7 @@ fil-ast.workspace = true
 fil-ir.workspace = true
 fil-utils.workspace = true
 fil-gen.workspace = true
+ordered-float.workspace = true
 
 calyx-ir.workspace = true
 calyx-frontend.workspace = true

crates/filament/src/ast_passes/mod.rs

@@ -1,3 +1,5 @@
+mod scheduling_model;
 mod toplevel;
 
+pub use scheduling_model::SchedulingModel;
 pub use toplevel::TopLevel;

crates/filament/src/ast_passes/scheduling_model.rs

@@ -0,0 +1,74 @@
+use crate::ast_visitor::{Action, Construct, Visitor};
+use fil_ast as ast;
+use fil_utils::{self as utils, AttrCtx, GPosIdx};
+use serde::Deserialize;
+use std::{collections::HashMap, fs};
+
+#[derive(Deserialize, Debug, Default)]
+struct CompModel {
+    ports: HashMap<String, f64>,
+    combinational: bool,
+}
+
+type Model = HashMap<String, CompModel>;
+
+/// Loads the scheduling model into attributes
+pub struct SchedulingModel {
+    /// Scheduling model mapping
+    model: Model,
+}
+
+impl Construct for SchedulingModel {
+    fn from(opts: &crate::cmdline::Opts, _: &mut fil_ast::Namespace) -> Self {
+        let model = opts
+            .scheduling_model
+            .as_ref()
+            .map(|path| {
+                toml::from_str(&fs::read_to_string(path).unwrap()).unwrap()
+            })
+            .unwrap_or_default();
+
+        Self { model }
+    }
+
+    fn clear_data(&mut self) {}
+}
+
+impl Visitor for SchedulingModel {
+    fn name() -> &'static str {
+        "scheduling_model"
+    }
+    fn signature(&mut self, sig: &mut ast::Signature) -> Action {
+        let comp = sig.name.inner().to_string();
+
+        log::info!(
+            "Scheduling model for component {}: {:?}",
+            comp,
+            self.model.get(&comp)
+        );
+
+        if let Some(attr_map) = self.model.get(&comp) {
+            if attr_map.combinational {
+                sig.attributes.set(
+                    utils::CompBool::Combinational,
+                    true,
+                    GPosIdx::UNKNOWN,
+                );
+            }
+
+            for port in sig.outputs_mut() {
+                let name = port.name.to_string();
+
+                let value = *attr_map.ports.get(&name).unwrap_or_else(|| panic!("Port {} not found in scheduling model for component {}",
+                    name, comp));
+
+                port.attrs.set(
+                    utils::PortFloat::CombDelay,
+                    value,
+                    GPosIdx::UNKNOWN,
+                );
+            }
+        }
+        Action::Continue
+    }
+}

crates/filament/src/cmdline.rs

@@ -110,6 +110,10 @@ pub struct Opts {
     #[argh(switch, long = "preserve-names")]
     pub preserve_names: bool,
 
+    /// scheduling model containing port delays relative to clock frequency
+    #[argh(option, long = "scheduling-model")]
+    pub scheduling_model: Option<String>,
+
     // Solver specific configuration
     /// solver to use (default: cvc5): cvc5, z3
     #[argh(option, long = "solver", default = "Solver::Z3")]