kaio 0.0.3

Rust-native GPU kernel authoring framework. Write GPU compute kernels in Rust, compile to PTX. Cross-platform (Windows + Linux), compile-time PTX emission, type-safe.
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

//! # KAIO
//!
//! **Rust-native GPU kernel authoring framework.**
//!
//! KAIO (πῦρ — fire) lets developers write GPU compute kernels in Rust and
//! compile them to PTX for execution on NVIDIA GPUs. It is a Rust alternative
//! to OpenAI's Triton, targeting Windows and Linux from day one, with
//! compile-time PTX emission and Rust's type-safety guarantees.
//!
//! ## Crates
//!
//! - [`kaio_core`] — PTX IR types, instruction emitters, PtxWriter
//! - [`kaio_runtime`] — CUDA device management, buffers, PTX loading, kernel launch
//!
//! ## Status
//!
//! **Phase 2 in progress.** `#[gpu_kernel]` proc macro transforms Rust
//! function syntax into PTX codegen + typed launch wrappers.

#![warn(missing_docs)]

pub mod gpu_builtins;
pub mod prelude;

/// PTX code generation — IR types, instruction emitters, and PTX text emission.
pub use kaio_core as core;

/// CUDA runtime — device management, buffers, PTX loading, kernel launch.
pub use kaio_runtime as runtime;

/// Re-export the `#[gpu_kernel]` attribute macro.
pub use kaio_macros::gpu_kernel;