mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
9563a5fee4
* add support for conv transpose 2d and add bench mark for float types * update bench calculation * enable testing all conv operations on metal
11 lines
247 B
Rust
11 lines
247 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,
|
|
);
|