moirai_gpu/lib.rs
1//! GPU launch-shape planning for Moirai.
2//!
3//! `moirai-gpu` owns the occupancy planner of atlas ADR 0002: it intersects a
4//! themis [`GpuTopology`](themis::GpuTopology) with a mnemosyne
5//! [`KernelResourceBudget`] into a [`LaunchShape`]. Device acquisition,
6//! buffers, transfers, kernel dispatch, and synchronization belong to the
7//! Hephaestus providers; Moirai does not depend on them (ADR 0041). A device
8//! operation is scheduled on Moirai as an ordinary task.
9
10#![forbid(unsafe_code)]
11#![deny(missing_docs)]
12
13pub mod occupancy;
14
15pub use mnemosyne_core::KernelResourceBudget;
16pub use occupancy::{LaunchShape, plan_launch, plan_persistent_launch, resident_blocks};
17
18/// Convenient imports for Moirai GPU planning consumers.
19pub mod prelude {
20 pub use crate::{
21 KernelResourceBudget, LaunchShape, plan_launch, plan_persistent_launch, resident_blocks,
22 };
23}