Struct GroupedGemmJob 

pub struct GroupedGemmJob { /* private fields */ }

One column-major matrix multiply in a shared-buffer grouped GEMM.

The three offsets address element positions in the caller-owned flat buffers. The left block has shape rows x contracted, the right block has shape contracted x cols, and the output block has shape rows x cols. Jobs may share either input block, but output spans must be disjoint because the operation has no reduction mode.

Examples

use tensor4all_tensorbackend::GroupedGemmJob;

let job = GroupedGemmJob::new(8, 4, 0, 2, 3, 5);
assert_eq!(job.out_offset(), 8);
assert_eq!(job.lhs_offset(), 4);
assert_eq!(job.rhs_offset(), 0);
assert_eq!((job.rows(), job.contracted(), job.cols()), (2, 3, 5));

Implementations

impl GroupedGemmJob

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.

pub fn out_offset(&self) -> usize

Return the output element offset.

pub fn lhs_offset(&self) -> usize

Return the left-input element offset.

pub fn rhs_offset(&self) -> usize

Return the right-input element offset.

pub fn rows(&self) -> usize

Return the output row count.

pub fn contracted(&self) -> usize

Return the contracted dimension.

pub fn cols(&self) -> usize

Return the output column count.

Trait Implementations

impl Clone for GroupedGemmJob

fn clone(&self) -> GroupedGemmJob

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for GroupedGemmJob

impl Debug for GroupedGemmJob

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Eq for GroupedGemmJob

impl Hash for GroupedGemmJob

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for GroupedGemmJob

fn eq(&self, other: &GroupedGemmJob) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for GroupedGemmJob