Skip to main content

aisimulate_core/engine/generalized/
mod.rs

1// SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2// SPDX-License-Identifier: Apache-2.0
3
4//! Runtime-neutral mock-engine composition.
5//!
6//! [`RankEngine`] is the boundary implemented by one scheduler/KV/timing core.
7//! [`GeneralizedMockerEngine`] composes one or more of those cores into one
8//! logical worker. A one-rank engine and an attention-DP engine therefore have
9//! the same caller-facing contract.
10//!
11//! Starting a pass eagerly commits every ready rank's work. Completion effects
12//! remain hidden until [`GeneralizedMockerEngine::complete_pass`] is called.
13//! With attention DP, all sibling ranks share the slowest rank's completion
14//! boundary, including ranks that had no work when the pass started.
15
16mod contracts;
17mod engine;
18
19pub use contracts::*;
20pub use engine::*;
21
22#[cfg(test)]
23mod tests;