mirror of https://github.com/tracel-ai/burn.git
changed benchmark
This commit is contained in:
parent
eac36e6eb7
commit
5250853936
|
@ -53,9 +53,9 @@ fn bench<B: Backend>(
|
|||
) {
|
||||
const D: usize = 3;
|
||||
let batch_size = 32;
|
||||
let m = 256;
|
||||
let k = 256;
|
||||
let n = 256;
|
||||
let m = 1024;
|
||||
let k = 1024;
|
||||
let n = 1024;
|
||||
let shape_lhs = [batch_size, m, k].into();
|
||||
let shape_rhs = [batch_size, k, n].into();
|
||||
|
||||
|
|
|
@ -33,8 +33,8 @@ pub enum MatmulStrategy {
|
|||
#[cfg(feature = "autotune")]
|
||||
impl Default for MatmulStrategy {
|
||||
fn default() -> Self {
|
||||
// MatmulStrategy::Tiling2d(Tiling2dConfig::default())
|
||||
MatmulStrategy::Cmma
|
||||
MatmulStrategy::Tiling2d(Tiling2dConfig::default())
|
||||
// MatmulStrategy::Cmma
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue