pub struct MoeExpertStore {
pub layers: Vec<LayerMoeWeights>,
}Expand description
Host-side expert weights for all MoE layers (migration source of truth).
Fields§
§layers: Vec<LayerMoeWeights>Implementations§
Source§impl MoeExpertStore
impl MoeExpertStore
pub fn num_layers(&self) -> usize
Sourcepub fn refresh_pools(
&self,
pools: &mut [ExpertPool],
captured: &[Vec<u32>],
decode_step: usize,
is_prefill_block: bool,
) -> bool
pub fn refresh_pools( &self, pools: &mut [ExpertPool], captured: &[Vec<u32>], decode_step: usize, is_prefill_block: bool, ) -> bool
Apply captured TopK indices to per-layer pools (TIDE refresh).
Sourcepub fn apply_to_compiled(&self, compiled: &mut CompiledGraph)
pub fn apply_to_compiled(&self, compiled: &mut CompiledGraph)
Push full host stacks into compiled params (lossless; refreshes arena bytes).
Trait Implementations§
Source§impl Clone for MoeExpertStore
impl Clone for MoeExpertStore
Source§fn clone(&self) -> MoeExpertStore
fn clone(&self) -> MoeExpertStore
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for MoeExpertStore
impl RefUnwindSafe for MoeExpertStore
impl Send for MoeExpertStore
impl Sync for MoeExpertStore
impl Unpin for MoeExpertStore
impl UnsafeUnpin for MoeExpertStore
impl UnwindSafe for MoeExpertStore
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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