pub struct ChunkedPrefillExecutor { /* private fields */ }Expand description
Executor for chunked prefill operations
Implementations§
Source§impl ChunkedPrefillExecutor
impl ChunkedPrefillExecutor
Sourcepub fn new(
model_executor: Arc<dyn ModelExecutor + Send + Sync>,
config: ChunkedPrefillConfig,
) -> Self
pub fn new( model_executor: Arc<dyn ModelExecutor + Send + Sync>, config: ChunkedPrefillConfig, ) -> Self
Create new chunked prefill executor
Sourcepub fn with_defaults(
model_executor: Arc<dyn ModelExecutor + Send + Sync>,
) -> Self
pub fn with_defaults( model_executor: Arc<dyn ModelExecutor + Send + Sync>, ) -> Self
Create with default config
Sourcepub async fn execute(&self, tokens: Vec<TokenId>) -> Result<PrefillOutput>
pub async fn execute(&self, tokens: Vec<TokenId>) -> Result<PrefillOutput>
Execute prefill with chunking if needed
Sourcepub fn config(&self) -> &ChunkedPrefillConfig
pub fn config(&self) -> &ChunkedPrefillConfig
Get configuration
Trait Implementations§
Auto Trait Implementations§
impl !RefUnwindSafe for ChunkedPrefillExecutor
impl !UnwindSafe for ChunkedPrefillExecutor
impl Freeze for ChunkedPrefillExecutor
impl Send for ChunkedPrefillExecutor
impl Sync for ChunkedPrefillExecutor
impl Unpin for ChunkedPrefillExecutor
impl UnsafeUnpin for ChunkedPrefillExecutor
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<T> ErasedDestructor for Twhere
T: 'static,
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