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

Broadcasting performance optimization (cpu) (#182)

Browse Source 
* Avoid recomputing the index from scratch each time.

* More performance optimisations.
This commit is contained in:
Laurent Mazare
2023-07-17 13:41:09 +01:00
committed by GitHub
parent 5b1c0bc9be
commit acb2f90469
4 changed files with 125 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -46,7 +46,7 @@ mod dtype;
mod dummy_cuda_backend;
mod error;
mod indexer;
mod layout;
pub mod layout;
#[cfg(feature = "mkl")]
mod mkl;
pub mod npy;