morok-codegen 0.1.0-alpha.2

Backend code generation 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
26
27
28
29
30
31
32
33
34
35

//! Code generation for morok tensor operations.
//!
//! This crate provides backend-agnostic code generation infrastructure
//! for converting optimized UOp graphs into executable code.
//!
//! # Architecture
//!
//! - **Traits**: Backend-agnostic interfaces (`Renderer`)
//! - **LLVM**: LLVM IR code generation for CPU execution
//! - **Future**: CUDA, Metal, OpenCL renderers
//!
//! # Usage
//!
//! ```ignore
//! use morok_codegen::llvm;
//!
//! let kernel = llvm::text::render(&optimized_uop_graph, Some("kernel"))?;
//! ```

pub mod c;
pub mod common;
pub mod error;
pub mod llvm;
#[cfg(feature = "mlir")]
pub mod mlir;
pub mod traits;
pub mod types;

#[cfg(test)]
pub mod test;

pub use common::collect_buffers_and_vars;
pub use error::*;
pub use traits::*;
pub use types::*;