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

Optimize Tensor::new when called on nested Vec<..>. (#2927)

Browse Source 
* Optimize Tensor::new when called on nested Vec<..>.

* Improve performance.

* Similar flattening for the 4d case.

* More tweaks.

* Add some dummy test.
This commit is contained in:
Laurent Mazare
2025-04-28 09:19:45 +02:00
committed by GitHub
parent e3db30021f
commit e98754fc5a
5 changed files with 174 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/benches/bench_main.rs
View File

@ -4,11 +4,12 @@ use criterion::criterion_main;
criterion_main!(
benchmarks::affine::benches,
benchmarks::copy::benches,
benchmarks::conv_transpose2d::benches,
benchmarks::matmul::benches,
benchmarks::qmatmul::benches,
benchmarks::random::benches,
benchmarks::reduce::benches,
benchmarks::unary::benches,
benchmarks::where_cond::benches,
benchmarks::conv_transpose2d::benches,
benchmarks::qmatmul::benches,
benchmarks::unary::benches
);