Skip to main content

WriteSlicePlan

Struct WriteSlicePlan 

Source
pub struct WriteSlicePlan<T: DeviceRepr + Copy + 'static, const N: usize> { /* private fields */ }
Expand description

write_slice plan.

dest[start_0..end_0, ..., start_{N-1}..end_{N-1}] = source — assign (not accumulate). Drives Fuel team’s persistent KV-cache append.

When to use: in-place per-axis range write. Distinct from ScatterAddPlan (which accumulates per-index) and from PadPlan (which produces a larger output tensor). No backward — non-differentiable.

Dtypes: every byte-aligned element kind in baracuda’s element bank — f16, bf16, f32, F32Strict, f64, i32, i64, Bool, S8, U8, Fp8E4M3, Fp8E5M2, Complex32, Complex64. Plus nibble-packed S4 / U4 with the even-alignment constraint on the innermost axis. Bin (1-bit packed) is out of scope.

Shape limits: rank in [1, 8]; per-axis 0 ≤ start ≤ end ≤ dest_shape[d]; source_shape[d] = end - start.

Workspace: none.

Precision guarantee: deterministic, bit-stable, bit-exact (no arithmetic — pure memcpy / index + copy).

Implementations§

Source§

impl<T: DeviceRepr + Copy + 'static, const N: usize> WriteSlicePlan<T, N>

Source

pub fn select( _stream: &Stream, desc: &WriteSliceDescriptor<N>, _pref: PlanPreference, ) -> Result<Self>

Pick a kernel for desc. Validates rank, range bounds, source shape consistency, dtype coverage, and the nibble-axis-alignment constraint for S4 / U4. Detects the available fast path.

Source

pub fn can_implement(&self, args: &WriteSliceArgs<'_, T, N>) -> Result<()>

Validate args against the descriptor: shapes match, device buffers are large enough.

Source

pub fn workspace_size(&self) -> usize

Workspace size in bytes. Always 0.

Source

pub fn sku(&self) -> KernelSku

Identity of the kernel this plan picked.

Source

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees for this plan’s kernel.

Source

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: WriteSliceArgs<'_, T, N>, ) -> Result<()>

Launch on stream. workspace is ignored (always zero).

Auto Trait Implementations§

§

impl<T, const N: usize> Freeze for WriteSlicePlan<T, N>

§

impl<T, const N: usize> RefUnwindSafe for WriteSlicePlan<T, N>
where T: RefUnwindSafe,

§

impl<T, const N: usize> Send for WriteSlicePlan<T, N>
where T: Send,

§

impl<T, const N: usize> Sync for WriteSlicePlan<T, N>
where T: Sync,

§

impl<T, const N: usize> Unpin for WriteSlicePlan<T, N>
where T: Unpin,

§

impl<T, const N: usize> UnsafeUnpin for WriteSlicePlan<T, N>

§

impl<T, const N: usize> UnwindSafe for WriteSlicePlan<T, N>
where T: UnwindSafe,

Blanket Implementations§

Source§

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

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

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

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

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

Source§

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

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

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

Source§

fn into(self) -> U

Calls U::from(self).

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

Source§

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

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

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

Performs the conversion.
Source§

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

Source§

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

The type returned in the event of a conversion error.
Source§

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

Performs the conversion.