morok-ir 0.1.0-alpha.2

Intermediate representation for the Morok ML compiler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! Common imports for working with UOp graphs.
//!
//! This module provides a convenient way to import the most commonly used types
//! when working with the IR:
//!
//! ```rust,ignore
//! use morok_ir::prelude::*;
//! ```

// Core types
pub use crate::Op;
pub use crate::uop::{IntoUOp, UOp, UOpKey};

// Operation types
pub use crate::types::{BinaryOp, ConstValue, ConstValueHash, ReduceOp, TernaryOp, UnaryOp};

// Shape and indexing
pub use crate::indexing::IndexSpec;
pub use crate::sint::SInt;

// Re-exports from dependencies
pub use morok_dtype::DType;
pub use morok_dtype::DeviceSpec;

pub use strum::AsRefStr;