rublas 0.1.9

Linear algebra and matrix multiplication for Ruda.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

//! Native ACLNN matrix multiplication on CANN-owned tensors.
pub use ruda_driver_cann::CannError;
pub use ruda_driver_cann::tensor::{CannSession, CannTensor, DType};

pub fn matmul(
    a: &CannTensor,
    b: &CannTensor,
    output_dtype: DType,
    cube_math_type: i8,
) -> Result<CannTensor, CannError> {
    a.session().matmul(a, b, output_dtype, cube_math_type)
}