pub struct MixedBatchResult {
pub decode_logits: DevicePtr,
pub prefill_logits: Vec<DevicePtr>,
}Expand description
Result of a fully-batched mixed forward pass: M decode tokens + N prefill chunks in one pass.
Fields§
§decode_logits: DevicePtrLogits for decode lanes: [M, vocab] BF16. NULL if no decode lanes.
prefill_logits: Vec<DevicePtr>Logits per prefill stream — one DevicePtr per stream in the input
slice, in the same order. Each entry is [1, vocab] BF16 when that
stream’s chunk was is_last_chunk, or NULL otherwise.
Auto Trait Implementations§
impl Freeze for MixedBatchResult
impl RefUnwindSafe for MixedBatchResult
impl Send for MixedBatchResult
impl Sync for MixedBatchResult
impl Unpin for MixedBatchResult
impl UnwindSafe for MixedBatchResult
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