Skip to main content

Crate moirai_gpu

Crate moirai_gpu 

Source
Expand description

GPU launch-shape planning for Moirai.

moirai-gpu owns the occupancy planner of atlas ADR 0002: it intersects a themis GpuTopology with a mnemosyne KernelResourceBudget into a LaunchShape. Device acquisition, buffers, transfers, kernel dispatch, and synchronization belong to the Hephaestus providers; Moirai does not depend on them (ADR 0041). A device operation is scheduled on Moirai as an ordinary task.

Re-exports§

pub use occupancy::LaunchShape;
pub use occupancy::plan_launch;
pub use occupancy::plan_persistent_launch;
pub use occupancy::resident_blocks;

Modules§

occupancy
Warp-aware launch-shape planning (atlas ADR 0002, Stage D2).
prelude
Convenient imports for Moirai GPU planning consumers.

Structs§

KernelResourceBudget
A kernel’s per-launch resource requirements.