kaio-runtime 0.2.2

KAIO runtime — CUDA driver API wrapper, kernel launch, and device memory management. Part of the KAIO GPU kernel authoring framework.
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
36
37
38
39
40
41
42
43
44

//! # kaio-runtime
//!
//! CUDA runtime layer for the KAIO GPU kernel authoring framework. This
//! crate wraps [`cudarc`] to provide device management, typed device
//! buffers, PTX module loading, and a builder-style kernel launch API. It
//! is Layer 2 of KAIO, sitting on top of [`kaio-core`] (PTX emission).
//!
//! ## Quick start
//!
//! ```ignore
//! use kaio_runtime::{KaioDevice, GpuBuffer};
//!
//! let device = KaioDevice::new(0)?;
//! let buf = device.alloc_from(&[1.0f32, 2.0, 3.0])?;
//! let host = buf.to_host(&device)?;
//! assert_eq!(host, vec![1.0, 2.0, 3.0]);
//! ```
//!
//! ## GPU-gated tests
//!
//! Tests that require an NVIDIA GPU are `#[ignore]`-gated. Run them with:
//!
//! ```sh
//! cargo test -p kaio-runtime -- --ignored
//! ```
//!
//! Standard `cargo test --workspace` runs only host-side tests.
//!
//! [`cudarc`]: https://crates.io/crates/cudarc
//! [`kaio-core`]: https://crates.io/crates/kaio-core

#![warn(missing_docs)]

pub mod buffer;
pub mod device;
pub mod error;
pub mod module;

pub use buffer::GpuBuffer;
pub use cudarc::driver::LaunchConfig;
pub use cudarc::driver::PushKernelArg;
pub use device::{DeviceInfo, KaioDevice};
pub use error::{KaioError, Result};
pub use module::{KaioFunction, KaioModule};