Struct TensorMapArg 

pub struct TensorMapArg<K: TensorMapKind> {
    pub tensor: TensorArg,
    pub metadata: TensorMapMeta,
    pub _kind: PhantomData<K>,
}

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

metadata: TensorMapMeta

_kind: PhantomData<K>

Implementations

impl<K: TensorMapKind> TensorMapArg<K>

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

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

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

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

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

pub fn with_nan_fill(self) -> Self