pub struct PassShape {Show 14 fields
pub rows_per_sequence: u32,
pub emitted_per_sequence: f64,
pub weight_bytes: f64,
pub embed_bytes: f64,
pub read_bytes: f64,
pub write_bytes: f64,
pub rewrite_bytes: f64,
pub array_macs: f64,
pub vector_macs: f64,
pub scale_cycles: f64,
pub held_shared: f64,
pub held_per_sequence: f64,
pub on_chip_per_sequence: f64,
pub provenance: Provenance,
}Expand description
A pass as a function of its batch. Everything a pass costs is shared by the batch or paid by each sequence, so the cost is affine in the batch and in how the batch splits between the card’s memory and the host’s.
Fields§
§rows_per_sequence: u32Token-rows each sequence carries through a pass: its verified token and its drafts.
emitted_per_sequence: f64Tokens each sequence emits per pass, expected.
weight_bytes: f64Weight bytes a pass streams, whatever its batch.
embed_bytes: f64Embedding-table bytes a pass reads for each sequence: a row per verified row and per drafting step.
read_bytes: f64State, cache, log and ring bytes a pass reads and writes for each sequence.
write_bytes: f64§rewrite_bytes: f64Of those, the bytes of each sequence’s recurrent states read and written back in place; the rest are read in long runs with the pass’s rows appended after them.
array_macs: f64§vector_macs: f64§scale_cycles: f64Block-scale lane cycles a pass spends for each sequence.
Bytes every serving of the checkpoint holds: the weights and the embedding table.
held_per_sequence: f64Bytes each sequence holds: its states, its log, its ring, its caches.
on_chip_per_sequence: f64Bytes each sequence’s token-rows take on chip through a layer.
provenance: ProvenanceThe weakest acceptance the shape rests on.
Implementations§
Source§impl PassShape
impl PassShape
Sourcepub fn of(h: &GgufHeader, s: &Serving) -> Result<PassShape, ServingRefusal>
pub fn of(h: &GgufHeader, s: &Serving) -> Result<PassShape, ServingRefusal>
The shape of a pass of s over the checkpoint h describes. The batch and the host split are the
cost’s arguments; everything else in s is read here.
Sourcepub fn cost(
&self,
card: &Card,
batch: u32,
host_sequences: u32,
) -> Result<ServingCost, ServingRefusal>
pub fn cost( &self, card: &Card, batch: u32, host_sequences: u32, ) -> Result<ServingCost, ServingRefusal>
The pass of batch sequences, host_sequences of them held in host memory, on card.
Sourcepub fn cost_split(
&self,
card: &Card,
batch: u32,
host_memory: bool,
) -> Result<ServingCost, ServingRefusal>
pub fn cost_split( &self, card: &Card, batch: u32, host_memory: bool, ) -> Result<ServingCost, ServingRefusal>
The pass of batch sequences split between the card’s memory and the host’s as fast as the split allows:
the card’s memory streams far faster, so it holds sequences until its own traffic would outlast the host
link’s.
Sourcepub fn best(
&self,
card: &Card,
host_memory: bool,
) -> Result<ServingCost, ServingRefusal>
pub fn best( &self, card: &Card, host_memory: bool, ) -> Result<ServingCost, ServingRefusal>
The batch that makes card emit the most tokens per second, and its pass. Ties go to the smaller batch.
Trait Implementations§
impl StructuralPartialEq for PassShape
Auto Trait Implementations§
impl Freeze for PassShape
impl RefUnwindSafe for PassShape
impl Send for PassShape
impl Sync for PassShape
impl Unpin for PassShape
impl UnsafeUnpin for PassShape
impl UnwindSafe for PassShape
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can
then be further downcast into Box<ConcreteType> where ConcreteType implements Trait.§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be
further downcast into Rc<ConcreteType> where ConcreteType implements Trait.§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.