Skip to main content

vyre_runtime/megakernel/planner/
mod.rs

1//! Resident work-queue sizing, launch geometry, and work-item contracts.
2
3mod barriers;
4mod caps;
5mod config;
6mod geometry;
7mod grid;
8#[cfg(feature = "self-substrate-adapters")]
9mod programs;
10mod sizing;
11
12pub use barriers::{elide_value_flow_barriers, BarrierElisionReport};
13pub use caps::{
14    ResidentQueueCapabilities, ResidentQueueReport, ResidentQueueTelemetry, ResidentWorkItem,
15};
16pub use config::{ResidentQueueConfig, ResidentWorkloadHints};
17pub use geometry::{
18    default_worker_groups_from_limits, dispatch_grid_for, padded_slot_count, worker_workgroup_size,
19    ResidentLaunchGeometry,
20};
21pub use grid::{ResidentGridLimits, ResidentGridPlan, ResidentGridRequest};
22#[cfg(feature = "self-substrate-adapters")]
23pub use programs::{
24    build_bellman_tn_order_program, build_kfac_autotune_step_program,
25    build_persistent_fixpoint_program, build_scallop_provenance_wide_program,
26    build_sinkhorn_clustering_program, build_sinkhorn_full_clustering_program,
27};
28pub use sizing::ResidentSizingPolicy;
29
30#[cfg(test)]
31use super::task::TaskWorkItem;
32
33#[cfg(test)]
34#[path = "../core_tests.rs"]
35mod tests;