ChunkedPrefillPageMetadata

spark_model::traits

Struct ChunkedPrefillPageMetadata

pub struct ChunkedPrefillPageMetadata {
    pub block_table: DevicePtr,
    pub seq_len: DevicePtr,
    pub block_capacity: usize,
    pub uploaded_blocks: usize,
}

Expand description

Per-sequence paged attention metadata for chunked prefill.

Positions and slots remain chunk-local, but the paged block table and running sequence length can persist across chunks so we only upload the changed tail instead of rebuilding the full page metadata every time.

Fields§

§block_table: DevicePtr

Device buffer holding the sequence block table as raw 32-bit entries.

§seq_len: DevicePtr

Device buffer holding the running paged-prefill sequence length.

§block_capacity: usize

Total block-table entries allocated for this prompt.

§uploaded_blocks: usize

Number of block-table entries already uploaded to block_table.

Auto Trait Implementations§

impl Freeze for ChunkedPrefillPageMetadata

impl RefUnwindSafe for ChunkedPrefillPageMetadata

impl Send for ChunkedPrefillPageMetadata

impl Sync for ChunkedPrefillPageMetadata

impl Unpin for ChunkedPrefillPageMetadata

impl UnwindSafe for ChunkedPrefillPageMetadata

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more