Trait custos_math::Gemm

pub trait Gemm<T> {
    fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>;
}

Matrix multiplication. Uses provided device.

Example

use custos::{CPU, VecRead};
use custos_math::{Matrix, Gemm};

let device = CPU::new();

let a = Matrix::from((&device, (2, 3), [1., 2., 3., 4., 5., 6.,]));
let b = Matrix::from((&device, (3, 2), [6., 5., 4., 3., 2., 1.,]));

let c = device.gemm(&a, &b);

assert_eq!(device.read(c.as_buf()), vec![20., 14., 56., 41.,]);

Required Methods

fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Implementations on Foreign Types

impl<T: GenericBlas + Default + Copy> Gemm<T> for CPU

fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T: CDatatype> Gemm<T> for CLDevice

fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

impl<T: GenericBlas> Gemm<T> for CudaDevice

fn gemm(&self, lhs: &Matrix<'_, T>, rhs: &Matrix<'_, T>) -> Matrix<'_, T>

Implementors