Crate morok_codegen 

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

use morok_codegen::llvm;

let kernel = llvm::text::render(&optimized_uop_graph, Some("kernel"))?;

Re-exports

pub use common::collect_buffers_and_vars;

pub use error::*;

pub use traits::*;

pub use types::*;

Modules

c

C source code generation backend.

common

Common utilities shared between codegen backends.

error

Error types for code generation.

llvm

LLVM IR code generation.

traits

Core traits for code generation.

types

Types for code generation.