morok-schedule 0.1.0-alpha.2

Optimization passes and pattern engine for the Morok ML compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

//! Backend-agnostic IR transformation passes.
//!
//! This module contains transformations that normalize the IR before
//! code generation, eliminating backend-specific logic from codegen.
//!
//! # Passes
//!
//! - [`linearize_index`] - Multi-index → single linear offset normalization
//!
//! These passes run during post-optimization, after rangeify and before codegen.

pub mod linearize_index;

#[cfg(test)]
pub(crate) use linearize_index::pm_linearize_multi_index;
pub use linearize_index::{build_linear_index, compute_row_major_strides, count_divmod, extract_index_dimension};