pub struct MatMul { /* private fields */ }Expand description
Matrix multiplication kernel.
This is a batch kernel that multiplies two matrices. Used for benchmarking compute throughput.
Implementations§
Trait Implementations§
Source§impl BatchKernel<MatMulInput, MatMulOutput> for MatMul
impl BatchKernel<MatMulInput, MatMulOutput> for MatMul
Source§fn execute<'life0, 'async_trait>(
&'life0 self,
input: MatMulInput,
) -> Pin<Box<dyn Future<Output = Result<MatMulOutput>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
fn execute<'life0, 'async_trait>(
&'life0 self,
input: MatMulInput,
) -> Pin<Box<dyn Future<Output = Result<MatMulOutput>> + Send + 'async_trait>>where
Self: 'async_trait,
'life0: 'async_trait,
Execute the kernel with the given input. Read more
Source§fn validate_input(&self, input: &MatMulInput) -> Result<()>
fn validate_input(&self, input: &MatMulInput) -> Result<()>
Validate the input before execution. Read more
Source§fn execute_with_context<'life0, 'life1, 'async_trait>(
&'life0 self,
ctx: &'life1 ExecutionContext,
input: I,
) -> Pin<Box<dyn Future<Output = Result<O>> + Send + 'async_trait>>where
I: 'async_trait,
Self: Sync + 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
fn execute_with_context<'life0, 'life1, 'async_trait>(
&'life0 self,
ctx: &'life1 ExecutionContext,
input: I,
) -> Pin<Box<dyn Future<Output = Result<O>> + Send + 'async_trait>>where
I: 'async_trait,
Self: Sync + 'async_trait,
'life0: 'async_trait,
'life1: 'async_trait,
Execute the kernel with execution context. Read more
Source§impl GpuKernel for MatMul
impl GpuKernel for MatMul
Source§fn metadata(&self) -> &KernelMetadata
fn metadata(&self) -> &KernelMetadata
Returns the kernel metadata.
Source§fn requires_gpu_native(&self) -> bool
fn requires_gpu_native(&self) -> bool
Returns true if this kernel requires GPU-native execution.
Source§fn health_check(&self) -> HealthStatus
fn health_check(&self) -> HealthStatus
Perform a health check on this kernel. Read more
Source§fn refresh_config(&mut self, _config: &KernelConfig) -> Result<()>
fn refresh_config(&mut self, _config: &KernelConfig) -> Result<()>
Refresh kernel configuration at runtime. Read more
Auto Trait Implementations§
impl Freeze for MatMul
impl RefUnwindSafe for MatMul
impl Send for MatMul
impl Sync for MatMul
impl Unpin for MatMul
impl UnwindSafe for MatMul
Blanket Implementations§
Source§impl<T> ArchivePointee for T
impl<T> ArchivePointee for T
Source§type ArchivedMetadata = ()
type ArchivedMetadata = ()
The archived version of the pointer metadata for this type.
Source§fn pointer_metadata(
_: &<T as ArchivePointee>::ArchivedMetadata,
) -> <T as Pointee>::Metadata
fn pointer_metadata( _: &<T as ArchivePointee>::ArchivedMetadata, ) -> <T as Pointee>::Metadata
Converts some archived metadata to the pointer metadata for itself.
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<F, W, T, D> Deserialize<With<T, W>, D> for F
impl<F, W, T, D> Deserialize<With<T, W>, D> for F
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> LayoutRaw for T
impl<T> LayoutRaw for T
Source§fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
Gets the layout of the type.