oxicuda 0.1.2

OxiCUDA - Pure Rust CUDA replacement for the COOLJAPAN ecosystem (95% performance target)
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

//! ONNX GPU Inference Backend for OxiCUDA.
//!
//! Provides a complete ONNX operator runtime with:
//! - IR types for representing ONNX graphs ([`ir`](crate::onnx_backend::ir))
//! - 60+ operator implementations ([`ops`](crate::onnx_backend::ops))
//! - Graph executor with topological ordering ([`executor`](crate::onnx_backend::executor))
//! - Memory planning with buffer reuse ([`planner`](crate::onnx_backend::planner))
//! - Operator fusion passes ([`fusion`](crate::onnx_backend::fusion))
//! - Shape inference engine ([`shape_inference`](crate::onnx_backend::shape_inference))

pub mod executor;
pub mod fusion;
pub mod ir;
pub mod ops;
pub mod planner;
pub mod shape_inference;

pub use executor::GraphExecutor;
pub use fusion::{FusionPass, FusionResult};
pub use ir::{
    AttributeValue, DataType, Graph, Node, OnnxError, OnnxResult, OnnxTensor, TensorInfo,
    TensorShape,
};
pub use ops::OpRegistry;
pub use planner::{AllocationInfo, MemoryPlan, MemoryPlanner};
pub use shape_inference::ShapeInference;