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

WGPU: matmul vec4 (#897)

This commit is contained in:
Louis Fortier-Dubois 2023-10-24 17:23:43 -04:00 committed by GitHub
parent 0ab611b42e
commit e76b6d47de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
burn-wgpu/src/ops/float_ops.rs
View File

@ -1,4 +1,5 @@
use super::{numeric, BoolTensor, Device, FloatElem, FloatTensor, FullPrecisionBackend, IntTensor};
use crate::kernel::matmul::vec4_primitive::matmul_tiling_2d_vec4_primitive_default;
use crate::kernel::prng::{random_bernoulli, random_normal, random_uniform};
use crate::kernel::{
self, unary_default, unary_inplace_default, unary_scalar_default, unary_scalar_inplace_default,
@ -143,7 +144,7 @@ where
lhs: FloatTensor<Self, D>,
rhs: FloatTensor<Self, D>,
) -> FloatTensor<Self, D> {
kernel::matmul::contiguous::matmul_tiling_2d_default(lhs, rhs)
matmul_tiling_2d_vec4_primitive_default(lhs, rhs)
}
fn swap_dims<const D: usize>(

6
codecov.yml Normal file
View File

@ -0,0 +1,6 @@
coverage:
status:
project:
default:
target: 80%
threshold: 5%

2
examples/mnist/src/training.rs
View File

@ -3,7 +3,7 @@ use crate::model::Model;
use burn::module::Module;
use burn::optim::decay::WeightDecayConfig;
use burn::optim::{AdamConfig, AdamWConfig};
use burn::optim::AdamConfig;
use burn::record::{CompactRecorder, NoStdTrainingRecorder};
use burn::train::metric::store::{Aggregate, Direction, Split};
use burn::train::metric::{CpuMemory, CpuTemperature, CpuUse};