Skip to main content

TensorDispatchCtx

atomr_accel_cuda::kernel::dispatch

Struct TensorDispatchCtx

pub struct TensorDispatchCtx {
    pub handle: Arc<Mutex<SendHandle>>,
    pub stream: Arc<CudaStream>,
    pub completion: Arc<dyn CompletionStrategy>,
    pub plan_cache: Arc<PlanCache>,
    pub workspace: Arc<WorkspacePool>,
}

Expand description

TensorDispatch is owned by kernel::tensor (Phase 2 cuTENSOR).

Fields§

§handle: Arc<Mutex<SendHandle>>§stream: Arc<CudaStream>§completion: Arc<dyn CompletionStrategy>§plan_cache: Arc<PlanCache>§workspace: Arc<WorkspacePool>

Auto Trait Implementations§

impl Freeze for TensorDispatchCtx

impl !RefUnwindSafe for TensorDispatchCtx

impl Send for TensorDispatchCtx

impl Sync for TensorDispatchCtx

impl Unpin for TensorDispatchCtx

impl UnsafeUnpin for TensorDispatchCtx

impl !UnwindSafe for TensorDispatchCtx

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

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