huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

MKL adjustments. (#87)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/cuda_backend.rs
View File

@ -543,7 +543,7 @@ pub struct CudaStorage {
device: CudaDevice,
}
pub(crate) fn gemm_config<T>(
fn gemm_config<T>(
alpha: T,
beta: T,
(b, m, n, k): (usize, usize, usize, usize),