mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 02:58:50 +00:00
Add mkl support for matrix multiply. (#86)
* Fix some rebase issues. * Use mkl instead. * Use mkl in bert. * Add the optional mkl feature. * Conditional compilation based on the mkl feature. * Add more mkl support.
This commit is contained in:
@ -543,7 +543,7 @@ pub struct CudaStorage {
|
||||
device: CudaDevice,
|
||||
}
|
||||
|
||||
fn gemm_config<T>(
|
||||
pub(crate) fn gemm_config<T>(
|
||||
alpha: T,
|
||||
beta: T,
|
||||
(b, m, n, k): (usize, usize, usize, usize),
|
||||
|
Reference in New Issue
Block a user