pub struct PrefillLayout {
pub local: Contract,
pub wire: Contract,
pub arena_base: u64,
pub tp_size: i64,
pub tp_rank: i64,
pub pp_rank: i64,
pub fields_by_stage: Vec<Vec<String>>,
}Expand description
What a Prefill rank plans its routes and copies with.
Fields§
§local: ContractThis rank’s contract, with its landing addresses: the stage’s resident fields only under a pipeline.
wire: ContractThe logical contract every rank registers with the bootstrap server: the complete model’s, without addresses.
arena_base: u64This rank’s arena base.
tp_size: i64The attention TP size within a stage.
tp_rank: i64This rank’s attention TP rank.
pp_rank: i64This rank’s pipeline stage.
fields_by_stage: Vec<Vec<String>>Every stage’s resident fields.
Trait Implementations§
Source§impl Clone for PrefillLayout
impl Clone for PrefillLayout
Auto Trait Implementations§
impl Freeze for PrefillLayout
impl RefUnwindSafe for PrefillLayout
impl Send for PrefillLayout
impl Sync for PrefillLayout
impl Unpin for PrefillLayout
impl UnsafeUnpin for PrefillLayout
impl UnwindSafe for PrefillLayout
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