pub trait BLAS3<C: BLASContext>: Sized + Copy {
// Required method
unsafe fn gemm(
ctx: C,
transa: MatrixOp,
transb: MatrixOp,
m: i32,
n: i32,
k: i32,
alpha: Self,
a: DPtr<Self, C::Device>,
lda: i32,
b: DPtr<Self, C::Device>,
ldb: i32,
beta: Self,
c: DPtr<Self, C::Device>,
ldc: i32,
);
}
Expand description
BLAS Level 3 operations (Matrix-Matrix)
Required Methods§
Sourceunsafe fn gemm(
ctx: C,
transa: MatrixOp,
transb: MatrixOp,
m: i32,
n: i32,
k: i32,
alpha: Self,
a: DPtr<Self, C::Device>,
lda: i32,
b: DPtr<Self, C::Device>,
ldb: i32,
beta: Self,
c: DPtr<Self, C::Device>,
ldc: i32,
)
unsafe fn gemm( ctx: C, transa: MatrixOp, transb: MatrixOp, m: i32, n: i32, k: i32, alpha: Self, a: DPtr<Self, C::Device>, lda: i32, b: DPtr<Self, C::Device>, ldb: i32, beta: Self, c: DPtr<Self, C::Device>, ldc: i32, )
Compute the Generalised Matrix-Matrix multiplication:
C = alpha * AB + beta * C
§Safety
This is often a call across an FFI barrier, so the links or devices need to be running and may perform UB unchecked by rust
Dyn Compatibility§
This trait is not dyn compatible.
In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.