Skip to main content

TensorMapArg

cubecl_core::frontend

Struct TensorMapArg 

Source 
pub struct TensorMapArg<R: Runtime, K: TensorMapKind> {
    pub tensor: TensorArg<R>,
    pub metadata: TensorMapMeta,
    pub _kind: PhantomData<K>,
}
Expand description

Grid constant tensor map, currently only maps to CUDA tensormap. May be interleaved or swizzled, but last dimension must be contiguous (since strides don’t include the last dimension).

The tensormap is treated as an opaque type at runtime.

Fields§

§tensor: TensorArg<R>§metadata: TensorMapMeta§_kind: PhantomData<K>

Implementations§

Source§

impl<R: Runtime, K: TensorMapKind> TensorMapArg<R, K>

Source

pub fn new( args: K::Args, tensor: TensorArg<R>, storage_ty: impl Into<ElemType>, ) -> Self

Source

pub fn with_elem_stride(self, elem_stride: Strides) -> Self

Source

pub fn with_interleave(self, interleave: TensorMapInterleave) -> Self

Source

pub fn with_swizzle(self, swizzle: TensorMapSwizzle) -> Self

Source

pub fn with_prefetch(self, prefetch: TensorMapPrefetch) -> Self

Source

pub fn with_nan_fill(self) -> Self

Auto Trait Implementations§

§

impl<R, K> !RefUnwindSafe for TensorMapArg<R, K>

§

impl<R, K> !UnwindSafe for TensorMapArg<R, K>

§

impl<R, K> Freeze for TensorMapArg<R, K>
where TensorArg<R>: Freeze, PhantomData<K>: Freeze,

§

impl<R, K> Send for TensorMapArg<R, K>
where TensorArg<R>: Send, PhantomData<K>: Send,

§

impl<R, K> Sync for TensorMapArg<R, K>
where TensorArg<R>: Sync, PhantomData<K>: Sync,

§

impl<R, K> Unpin for TensorMapArg<R, K>
where TensorArg<R>: Unpin, PhantomData<K>: Unpin,

§

impl<R, K> UnsafeUnpin for TensorMapArg<R, K>
where TensorArg<R>: UnsafeUnpin, PhantomData<K>: UnsafeUnpin,

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<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> Downcast for T
where T: Any,

Source§

fn into_any(self: Box<T>) -> Box<dyn Any>

Converts Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>, which can then be downcast into Box<dyn ConcreteType> where ConcreteType implements Trait.
Source§

fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>

Converts Rc<Trait> (where Trait: Downcast) to Rc<Any>, which can then be further downcast into Rc<ConcreteType> where ConcreteType implements Trait.
Source§

fn as_any(&self) -> &(dyn Any + 'static)

Converts &Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &Any’s vtable from &Trait’s.
Source§

fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)

Converts &mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot generate &mut Any’s vtable from &mut Trait’s.
Source§

impl<T> DowncastSend for T
where T: Any + Send,

Source§

fn into_any_send(self: Box<T>) -> Box<dyn Any + Send>

Converts Box<Trait> (where Trait: DowncastSend) to Box<dyn Any + Send>, which can then be downcast into Box<ConcreteType> where ConcreteType implements Trait.
Source§

impl<T> DowncastSync for T
where T: Any + Send + Sync,

Source§

fn into_any_sync(self: Box<T>) -> Box<dyn Any + Sync + Send>

Converts Box<Trait> (where Trait: DowncastSync) to Box<dyn Any + Send + Sync>, which can then be downcast into Box<ConcreteType> where ConcreteType implements Trait.
Source§

fn into_any_arc(self: Arc<T>) -> Arc<dyn Any + Sync + Send> ⓘ

Converts Arc<Trait> (where Trait: DowncastSync) to Arc<Any>, which can then be downcast into Arc<ConcreteType> where ConcreteType implements Trait.
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> IntoComptime for T

Source§

fn comptime(self) -> Self

Source§

impl<T> IntoEither for T

Source§

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 more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
where F: FnOnce(&Self) -> bool,

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
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

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

Source§

type Error = !

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.