pub struct PipelinePlan {
pub stages: Vec<StagePlacement>,
}Expand description
Pipeline execution plan for one model.
Fields§
§stages: Vec<StagePlacement>Implementations§
Source§impl PipelinePlan
impl PipelinePlan
Sourcepub fn from_assignments(
assignments: &[LayerAssignment],
device_by_node: &HashMap<String, DeviceKind>,
) -> Self
pub fn from_assignments( assignments: &[LayerAssignment], device_by_node: &HashMap<String, DeviceKind>, ) -> Self
Build a pipeline plan from layer assignments and a node/device map.
Sourcepub fn from_assignments_with_measured(
assignments: &[LayerAssignment],
device_by_node: &HashMap<String, DeviceKind>,
cluster: &ClusterState,
) -> Self
pub fn from_assignments_with_measured( assignments: &[LayerAssignment], device_by_node: &HashMap<String, DeviceKind>, cluster: &ClusterState, ) -> Self
Build a pipeline plan using measured latencies from cluster metrics if available.
Sourcepub fn est_step_latency_ms(&self) -> f32
pub fn est_step_latency_ms(&self) -> f32
Return aggregate token-step latency estimate for one micro-batch.
pub fn summary(&self) -> String
Trait Implementations§
Source§impl Clone for PipelinePlan
impl Clone for PipelinePlan
Source§impl Debug for PipelinePlan
impl Debug for PipelinePlan
Source§impl Default for PipelinePlan
impl Default for PipelinePlan
Source§impl PartialEq for PipelinePlan
impl PartialEq for PipelinePlan
impl StructuralPartialEq for PipelinePlan
Auto Trait Implementations§
impl Freeze for PipelinePlan
impl RefUnwindSafe for PipelinePlan
impl Send for PipelinePlan
impl Sync for PipelinePlan
impl Unpin for PipelinePlan
impl UnsafeUnpin for PipelinePlan
impl UnwindSafe for PipelinePlan
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
Source§fn in_current_span(self) -> Instrumented<Self> ⓘ
fn in_current_span(self) -> Instrumented<Self> ⓘ
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more