Skip to content

Commit

Permalink
Removed old applier code for fmulapplier and faddapplier
Browse files Browse the repository at this point in the history
  • Loading branch information
Myocardinal committed Aug 30, 2024
1 parent de4105d commit 23abbef
Showing 1 changed file with 31 additions and 31 deletions.
62 changes: 31 additions & 31 deletions src/language.rs
Original file line number Diff line number Diff line change
Expand Up @@ -525,41 +525,41 @@ pub fn rules<A: Analysis<TnsrLang>>() -> Vec<Rewrite<TnsrLang, TnsrAnalysis>> {
// foo: Var,
// bar: Var,
// }

impl Applier<TnsrLang, TnsrAnalysis> for FMulApplier {
fn apply_one(&self, egraph: &mut EGraph<TnsrLang, TnsrAnalysis>, matched_id: Id, subst: &Subst, _: Option<&PatternAst<TnsrLang>>, _: Symbol) -> Vec<Id> {
let foo_scalar = egraph[subst[self.foo]].data.scalar;
let bar_scalar = egraph[subst[self.bar]].data.scalar;
let new_scalar = foo_scalar * bar_scalar;
let new_var = egraph.add(TnsrLang::Var(new_scalar.to_string().into()));
let new_node = egraph.add(TnsrLang::Input([new_var]));
if egraph.union(new_node, matched_id) {
vec![new_node, matched_id]
} else {
vec![]
}
}
}

//
// impl Applier<TnsrLang, TnsrAnalysis> for FMulApplier {
// fn apply_one(&self, egraph: &mut EGraph<TnsrLang, TnsrAnalysis>, matched_id: Id, subst: &Subst, _: Option<&PatternAst<TnsrLang>>, _: Symbol) -> Vec<Id> {
// let foo_scalar = egraph[subst[self.foo]].data.scalar;
// let bar_scalar = egraph[subst[self.bar]].data.scalar;
// let new_scalar = foo_scalar * bar_scalar;
// let new_var = egraph.add(TnsrLang::Var(new_scalar.to_string().into()));
// let new_node = egraph.add(TnsrLang::Input([new_var]));
// if egraph.union(new_node, matched_id) {
// vec![new_node, matched_id]
// } else {
// vec![]
// }
// }
// }
//
// struct FAddApplier {
// foo: Var,
// bar: Var,
// }

impl Applier<TnsrLang, TnsrAnalysis> for FAddApplier {
fn apply_one(&self, egraph: &mut EGraph<TnsrLang, TnsrAnalysis>, matched_id: Id, subst: &Subst, _: Option<&PatternAst<TnsrLang>>, _: Symbol) -> Vec<Id> {
let foo_scalar = egraph[subst[self.foo]].data.scalar;
let bar_scalar = egraph[subst[self.bar]].data.scalar;
let new_scalar = foo_scalar + bar_scalar;
let new_var = egraph.add(TnsrLang::Var(new_scalar.to_string().into()));
let new_node = egraph.add(TnsrLang::Input([new_var]));
if egraph.union(new_node, matched_id) {
vec![new_node, matched_id]
} else {
vec![]
}
}
}
//
// impl Applier<TnsrLang, TnsrAnalysis> for FAddApplier {
// fn apply_one(&self, egraph: &mut EGraph<TnsrLang, TnsrAnalysis>, matched_id: Id, subst: &Subst, _: Option<&PatternAst<TnsrLang>>, _: Symbol) -> Vec<Id> {
// let foo_scalar = egraph[subst[self.foo]].data.scalar;
// let bar_scalar = egraph[subst[self.bar]].data.scalar;
// let new_scalar = foo_scalar + bar_scalar;
// let new_var = egraph.add(TnsrLang::Var(new_scalar.to_string().into()));
// let new_node = egraph.add(TnsrLang::Input([new_var]));
// if egraph.union(new_node, matched_id) {
// vec![new_node, matched_id]
// } else {
// vec![]
// }
// }
// }

pub struct TnsrCost<'a> {
pub egraph: &'a EGraph<TnsrLang, TnsrAnalysis>,
Expand Down

0 comments on commit 23abbef

Please sign in to comment.