dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

changed benchmark

This commit is contained in:
louisfd 2024-07-15 13:04:51 -04:00
parent eac36e6eb7
commit 5250853936
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend-comparison/benches/matmul.rs
View File

@ -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();

4
crates/burn-jit/src/kernel/matmul/base.rs
View File

@ -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
}
}