tokitai-operator 0.1.0

//! Intermediate representation: `SemanticGraph`, DSL, scheduling.
//!
//! The IR is the layer between the user's `TensorStore` and the
//! planner's `ExecutionPlan`. It is structured as a graph of
//! `SemanticNode`s, each of which holds the input/output ids, the
//! op name, the required/provided contracts, and the layer
//! behavior.
//!
//! - `semantic` (P343) — `SemanticGraph`, `SemanticNode`, and the
//!   imperative API.
//! - `dsl` (P343) — the closure-based sugar `Tokitai::graph(|g| ...)`.
//! - `contract` — the IR-level contract and theory-requirement types.
//! - `layer` — `LayerKind` and the per-layer schedule actions.
//! - `schedule` — the abstract `ScheduleAction` enum.
//!
//! Public types: `SemanticGraph`, `SemanticNode`, `TensorHandle`,
//! `GraphBuilder`, `CompiledGraph`.
//!
pub mod contract;
pub mod dsl;
pub mod layer;
pub mod schedule;
pub mod semantic;

pub use contract::RewriteJustification;
pub use dsl::{CompiledGraph, GraphBuilder, TensorHandle, run_graph};
pub use layer::{Layer, LayerKind, LayerTransition};
pub use semantic::{SemanticGraph, SemanticNode};