51 lines
2 KiB
Rust
51 lines
2 KiB
Rust
|
//! Test-only constructors and harness type aliases for the `tests` tree.
|
||
|
|
//!
|
||
|
|
//! These helpers previously lived in the production modules
|
||
|
|
//! (`orchestration/run_fsm`, `staging::control`) but serve only test code.
|
||
|
|
//! Defining them here keeps them compiled under `#[cfg(test)]` only, so
|
||
|
|
//! production builds stay free of the dead-code warnings they would otherwise
|
||
|
|
//! trigger. Inherent methods resolve through their type, so existing call sites
|
||
|
|
//! such as `fsm::RunPlan::test_linear(..)` and
|
||
|
|
//! `stage::WeightSource::embedded_gguf(..)` work unchanged.
|
||
|
|
|
||
|
|
use crate::run_fsm::{NodeId as FsmNodeId, OrchestratorRun, RunId as FsmRunId, RunPlan, StageRef};
|
||
|
|
use crate::run_plan::{GgufSource, TokenizerSource};
|
||
|
|
use crate::staging::{DeviceHandle, StageController, WeightSource};
|
||
|
|
|
||
|
|
impl RunPlan {
|
||
|
|
/// Build a linear pipeline plan (stage 0 -> stage 1 -> ... -> last stage).
|
||
|
|
/// Test-only convenience over the public `RunPlan` fields.
|
||
|
|
pub(crate) fn test_linear(run_id: FsmRunId, stages: Vec<StageRef>) -> Self {
|
||
|
|
Self { run_id, stages }
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Node ids in declared stage order.
|
||
|
|
pub(crate) fn stage_nodes(&self) -> Vec<FsmNodeId> {
|
||
|
|
self.stages.iter().map(|stage| stage.node_id).collect()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl DeviceHandle {
|
||
|
|
/// A handle on the worker's current (first) generation.
|
||
|
|
pub(crate) fn new_current(id: u64) -> Self {
|
||
|
|
Self { generation: 1, id }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl WeightSource {
|
||
|
|
/// Weights carried by an embedded GGUF file with an embedded tokenizer.
|
||
|
|
pub(crate) fn embedded_gguf(model_id: impl Into<String>, path: impl Into<String>) -> Self {
|
||
|
|
Self::new(
|
||
|
|
model_id,
|
||
|
|
GgufSource::LocalPath(path.into()),
|
||
|
|
TokenizerSource::EmbeddedGguf,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Test alias for the orchestrator run core, retained for readable test prose.
|
||
|
|
pub(crate) type OrchestratorHarness = OrchestratorRun;
|
||
|
|
|
||
|
|
/// Test alias for the stage controller core, retained for readable test prose.
|
||
|
|
pub(crate) type StageControllerHarness = StageController;
|