Rust-related/luminal
1
0
Fork 1
mirror of https://github.com/jafioti/luminal.git synced 2026-06-01 21:49:47 +09:00
Code Issues Packages Projects Releases Wiki Activity

tweaks

Browse Source
This commit is contained in:
Joe Fioti
2026-02-21 14:24:40 -08:00
parent 0659b2a0bd
commit 9794146f65
1 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/hlir.rs
View File

@@ -1,6 +1,7 @@
use std::fmt::Display; use std::fmt::Display;
use std::{fmt::Debug, sync::Arc}; use std::{fmt::Debug, sync::Arc};
use crate::egglog_utils::api::{rule, set};
use crate::egglog_utils::{ use crate::egglog_utils::{
api::{Rule, SortDef, sort}, api::{Rule, SortDef, sort},
base::*, base::*,
@@ -16,28 +17,19 @@ use itertools::Itertools;
/// Matches the op, reads dtype from the named source field, and sets it on the op. /// Matches the op, reads dtype from the named source field, and sets it on the op.
fn dtype_propagation_rule(sort: &SortDef, dtype_source: &str) -> Rule { fn dtype_propagation_rule(sort: &SortDef, dtype_source: &str) -> Rule {
let (args, op_match) = sort.new_call(); let (args, op_match) = sort.new_call();
crate::egglog_utils::api::rule(crate::egglog_utils::api::set( rule(set(dtype(op_match), dtype(args[dtype_source].clone())))
dtype(op_match.clone()),
dtype(args[dtype_source].clone()),
))
} }
/// Helper: build a dtype-from-field rule (dtype comes directly from a field variable). /// Helper: build a dtype-from-field rule (dtype comes directly from a field variable).
fn dtype_from_field_rule(sort: &SortDef, dtype_field: &str) -> Rule { fn dtype_from_field_rule(sort: &SortDef, dtype_field: &str) -> Rule {
let (args, op_match) = sort.new_call(); let (args, op_match) = sort.new_call();
crate::egglog_utils::api::rule(crate::egglog_utils::api::set( rule(set(dtype(op_match), args[dtype_field].clone()))
dtype(op_match.clone()),
args[dtype_field].clone(),
))
} }
/// Helper: build a rule that sets a fixed dtype on an op. /// Helper: build a rule that sets a fixed dtype on an op.
fn dtype_fixed_rule(sort: &SortDef, dtype_sort: &SortDef) -> Rule { fn dtype_fixed_rule(sort: &SortDef, dtype_sort: &SortDef) -> Rule {
let (_, op_match) = sort.new_call(); let (_, op_match) = sort.new_call();
crate::egglog_utils::api::rule(crate::egglog_utils::api::set( rule(set(dtype(op_match), dtype_sort.call(())))
dtype(op_match),
dtype_sort.call(()),
))
} }
use num_traits::Float; use num_traits::Float;
use petgraph::{Direction, algo::toposort, prelude::StableGraph, visit::EdgeRef}; use petgraph::{Direction, algo::toposort, prelude::StableGraph, visit::EdgeRef};