mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
0067fe00a8
* add basic unary bench for sqrt * process unary commands in tiles of 4 * re-enable all benchmarks * rename helper to unary * modify approach to split up tiled and non-tiled operations * undo bench ignore for other tests * update tile size to 2 * only perform the optimization on the contiguous even numbered element case
13 lines
312 B
Rust
13 lines
312 B
Rust
mod benchmarks;
|
|
|
|
use criterion::criterion_main;
|
|
criterion_main!(
|
|
benchmarks::affine::benches,
|
|
benchmarks::matmul::benches,
|
|
benchmarks::random::benches,
|
|
benchmarks::where_cond::benches,
|
|
benchmarks::conv_transpose2d::benches,
|
|
benchmarks::qmatmul::benches,
|
|
benchmarks::unary::benches
|
|
);
|