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

Fix two cuda bugs (matmul and where_cond).

Browse Source
This commit is contained in:
laurent
2023-06-27 11:31:04 +01:00
parent d7f729fb8f
commit 380d61e990
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -301,8 +301,8 @@ fn gemm_config<T>(
Ok(StridedBatchedConfig {
batch_size: b as i32,
gemm,
stride_a: (m * k) as i64,
stride_b: (n * k) as i64,
stride_a: (n * k) as i64,
stride_b: (m * k) as i64,
stride_c: (m * n) as i64,
})
}