Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Migrate/jit/matmul tiling 2d #1472

Merged
merged 28 commits into from
Mar 22, 2024
Merged

Migrate/jit/matmul tiling 2d #1472

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
83fd099
refactor matmul files
louisfd Mar 11, 2024
d0b2ce4
wip refactor matmul
louisfd Mar 11, 2024
603df3f
everything is memco
louisfd Mar 12, 2024
423d23d
support local arrays
louisfd Mar 12, 2024
fa222a0
advancing tiling2d
louisfd Mar 12, 2024
4419f45
advancing tiling2d
louisfd Mar 13, 2024
9b6f2d1
advancing tiling2d
louisfd Mar 13, 2024
feddabc
tiling2d finished but buggy
louisfd Mar 13, 2024
7e07825
configurable unrolling
louisfd Mar 13, 2024
855a77c
not bugged
louisfd Mar 13, 2024
11f55fd
fails on unroll
louisfd Mar 13, 2024
d045fe6
stupid break
louisfd Mar 13, 2024
a0bb940
tiling2d no assumption works
louisfd Mar 14, 2024
5596e3b
Merge branch 'main' into migrate/jit/matmul_tiling_2d
louisfd Mar 14, 2024
4123842
clippy
louisfd Mar 14, 2024
ed18cfa
bounds check as bool
louisfd Mar 19, 2024
651010a
lhs rhs as enum
louisfd Mar 19, 2024
5e8786b
tiling 2d major refactor
louisfd Mar 19, 2024
ea82414
remove assign vec4
louisfd Mar 19, 2024
8b504f7
variable declarations above loops
louisfd Mar 19, 2024
947c473
fmt
louisfd Mar 19, 2024
4c97c75
clippy
louisfd Mar 19, 2024
66076db
Fix autotune + unroll
nathanielsimard Mar 20, 2024
7a6299a
move val
louisfd Mar 21, 2024
1e9c815
Merge branch 'migrate/jit/matmul_tiling_2d' of github.com:tracel-ai/b…
louisfd Mar 21, 2024
b3af965
clippy
louisfd Mar 21, 2024
6237478
fmt
louisfd Mar 21, 2024
94a81ee
Merge branch 'main' into migrate/jit/matmul_tiling_2d
louisfd Mar 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix autotune + unroll
  • Loading branch information
@nathanielsimard
nathanielsimard committed Mar 20, 2024
commit 66076dbe0f721d84c368b210fa87a12b2aac8ee2
5 changes: 1 addition & 4 deletions crates/burn-jit/src/kernel/matmul/base.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -98,12 +98,9 @@
}

#[cfg(feature = "autotune")]
impl Default for MatmulStrategy {

Check failure on line 101 in crates/burn-jit/src/kernel/matmul/base.rs

View workflow job for this annotation

GitHub Actions / clippy

[clippy] crates/burn-jit/src/kernel/matmul/base.rs#L101 

error: this `impl` can be derived
   --> crates/burn-jit/src/kernel/matmul/base.rs:101:1
    |
101 | / impl Default for MatmulStrategy {
102 | |     fn default() -> Self {
103 | |         MatmulStrategy::Autotune
104 | |     }
105 | | }
    | |_^
    |
    = help: for further information visit https://rust-lang.github.io/rust-clippy/master/index.html#derivable_impls
    = note: `-D clippy::derivable-impls` implied by `-D warnings`
    = help: to override `-D warnings` add `#[allow(clippy::derivable_impls)]`
    = help: remove the manual implementation...
help: ...and instead derive it...
    |
83  + #[derive(Default)]
84  | pub enum MatmulStrategy {
    |
help: ...and mark the default variant
    |
97  ~     #[default]
98  ~     Autotune,
    |
Raw output 
crates/burn-jit/src/kernel/matmul/base.rs:101:1:e:error: this `impl` can be derived
   --> crates/burn-jit/src/kernel/matmul/base.rs:101:1
    |
101 | / impl Default for MatmulStrategy {
102 | |     fn default() -> Self {
103 | |         MatmulStrategy::Autotune
104 | |     }
105 | | }
    | |_^
    |
    = help: for further information visit https://rust-lang.github.io/rust-clippy/master/index.html#derivable_impls
    = note: `-D clippy::derivable-impls` implied by `-D warnings`
    = help: to override `-D warnings` add `#[allow(clippy::derivable_impls)]`
    = help: remove the manual implementation...
help: ...and instead derive it...
    |
83  + #[derive(Default)]
84  | pub enum MatmulStrategy {
    |
help: ...and mark the default variant
    |
97  ~     #[default]
98  ~     Autotune,
    |


__END__
fn default() -> Self {
MatmulStrategy::Simple {
grid_x: 32,
grid_y: 32,
}
MatmulStrategy::Autotune
}
}

Expand Down
2 changes: 1 addition & 1 deletion crates/burn-jit/src/kernel/matmul/tiling2d.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ impl<R: Runtime> DynamicKernelSource for MatmulTiling2dEagerKernel<R> {
variables: gpu::BinaryOperator { lhs, rhs, out },
config: self.config.clone(),
bounds_check_required: self.bounds_check_required,
unroll: false,
unroll: true,
}
.expand(&mut scope);

Expand Down
Loading