pub struct GroupedGemmJob { /* private fields */ }Expand description
One matrix multiply in a grouped GEMM over shared flat buffers.
Offsets are element offsets into the corresponding shared lhs, rhs, and
output buffers. Each job computes a column-major rows x cols output block
from a column-major rows x contracted lhs block and a column-major
contracted x cols rhs block.
Provider implementations receive these descriptors through the public grouped-GEMM request accessor. The engine validates ranges and pairwise output disjointness before provider entry.
Implementations§
Source§impl GroupedGemmJob
impl GroupedGemmJob
Sourcepub fn new(
out_offset: usize,
lhs_offset: usize,
rhs_offset: usize,
rows: usize,
contracted: usize,
cols: usize,
) -> Self
pub fn new( out_offset: usize, lhs_offset: usize, rhs_offset: usize, rows: usize, contracted: usize, cols: usize, ) -> Self
Construct a column-major grouped-GEMM job over shared flat buffers.
Sourcepub fn out_offset(&self) -> usize
pub fn out_offset(&self) -> usize
Return the output element offset.
Sourcepub fn lhs_offset(&self) -> usize
pub fn lhs_offset(&self) -> usize
Return the left-input element offset.
Sourcepub fn rhs_offset(&self) -> usize
pub fn rhs_offset(&self) -> usize
Return the right-input element offset.
Sourcepub fn contracted(&self) -> usize
pub fn contracted(&self) -> usize
Return the contracted dimension.
Trait Implementations§
Source§impl Clone for GroupedGemmJob
impl Clone for GroupedGemmJob
Source§fn clone(&self) -> GroupedGemmJob
fn clone(&self) -> GroupedGemmJob
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for GroupedGemmJob
impl Debug for GroupedGemmJob
Source§impl Hash for GroupedGemmJob
impl Hash for GroupedGemmJob
Source§impl PartialEq for GroupedGemmJob
impl PartialEq for GroupedGemmJob
Source§fn eq(&self, other: &GroupedGemmJob) -> bool
fn eq(&self, other: &GroupedGemmJob) -> bool
Tests for
self and other values to be equal, and is used by ==.impl Copy for GroupedGemmJob
impl Eq for GroupedGemmJob
impl StructuralPartialEq for GroupedGemmJob
Auto Trait Implementations§
impl Freeze for GroupedGemmJob
impl RefUnwindSafe for GroupedGemmJob
impl Send for GroupedGemmJob
impl Sync for GroupedGemmJob
impl Unpin for GroupedGemmJob
impl UnsafeUnpin for GroupedGemmJob
impl UnwindSafe for GroupedGemmJob
Blanket Implementations§
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<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
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 moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
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