pub struct PlanRuntimePrefillOutput { /* private fields */ }Expand description
Tensor-free prefill output bound to one request and committed KV extent.
Implementations§
Source§impl PlanRuntimePrefillOutput
impl PlanRuntimePrefillOutput
pub fn intermediate( request_id: RequestId, committed_tokens: usize, kv_cache: Arc<dyn KvCacheHandle>, ) -> Self
pub fn final_logits( request_id: RequestId, committed_tokens: usize, logits: Vec<f32>, kv_cache: Arc<dyn KvCacheHandle>, ) -> Result<Self>
pub fn request_id(&self) -> &RequestId
pub const fn committed_tokens(&self) -> usize
pub fn product(&self) -> &PlanRuntimePrefillProduct
pub fn kv_cache(&self) -> &Arc<dyn KvCacheHandle> ⓘ
pub fn validate_for_completion( &self, expected_request_id: &RequestId, completed_chunk: PrefillChunk, vocabulary_size: usize, ) -> Result<()>
pub fn into_parts( self, ) -> (PlanRuntimePrefillAuthority, PlanRuntimePrefillProduct)
Trait Implementations§
Auto Trait Implementations§
impl !RefUnwindSafe for PlanRuntimePrefillOutput
impl !UnwindSafe for PlanRuntimePrefillOutput
impl Freeze for PlanRuntimePrefillOutput
impl Send for PlanRuntimePrefillOutput
impl Sync for PlanRuntimePrefillOutput
impl Unpin for PlanRuntimePrefillOutput
impl UnsafeUnpin for PlanRuntimePrefillOutput
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