MKL adjustments. (#87)

This commit is contained in:
Laurent Mazare
2023-07-06 11:37:27 +01:00
committed by GitHub
parent c297a50960
commit dd60bd84bb
4 changed files with 44 additions and 38 deletions

View File

@ -129,6 +129,13 @@ pub enum Error {
#[error("cannot broadcast {src_shape:?} to {dst_shape:?}")]
BroadcastIncompatibleShapes { src_shape: Shape, dst_shape: Shape },
#[error("matmul is only supported for contiguous tensors lstride: {lhs_stride:?} rstride: {rhs_stride:?} mnk: {mnk:?}")]
MatMulNonContiguous {
lhs_stride: Vec<usize>,
rhs_stride: Vec<usize>,
mnk: (usize, usize, usize),
},
}
pub type Result<T> = std::result::Result<T, Error>;