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.

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

Matmul + CubeCL Update #2551

Merged
merged 9 commits into from
Nov 29, 2024
Merged

Matmul + CubeCL Update #2551

merged 9 commits into from
Nov 29, 2024

Conversation

nathanielsimard
Copy link
Member

No description provided.

MatmulStrategy::Cube => {
let out = init_matmul_output::<R, E>(&lhs, &rhs);
let client = &lhs.client;
cubecl::linalg::matmul::launch_ref::<R, E>(
&Default::default(),
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Since default in cubecl is Accelerated, if you're on wgpu and have autotune disabled, it will fail?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Since a very recent PR launch_ref of matmul in CubeCL returns a Result, so we get an error if Accelerated is not available. That way we can have a fallback here

@nathanielsimard nathanielsimard changed the title Matmulupdate Matmul + CubeCL Update Nov 28, 2024
Copy link

codecov bot commented Nov 29, 2024

Codecov Report

Attention: Patch coverage is 49.25373% with 34 lines in your changes missing coverage. Please review.

Project coverage is 81.86%. Comparing base (a5624c1) to head (5e7c800).
Report is 1 commits behind head on main.

Files with missing lines Patch % Lines
backend-comparison/src/lib.rs 0.00% 19 Missing ⚠️
...it/src/kernel/conv/conv2d/gemm/homogeneous/base.rs 0.00% 6 Missing ⚠️
...n-jit/src/kernel/conv/conv2d/gemm/loader/im2col.rs 0.00% 4 Missing ⚠️
crates/burn-jit/src/kernel/matmul/base.rs 50.00% 2 Missing ⚠️
crates/burn-jit/src/kernel/conv/conv2d/im2col.rs 50.00% 1 Missing ⚠️
crates/burn-jit/src/kernel/matmul/tune/base.rs 96.29% 1 Missing ⚠️
crates/burn-jit/src/ops/float_ops.rs 0.00% 1 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2551      +/-   ##
==========================================
+ Coverage   81.85%   81.86%   +0.01%     
==========================================
  Files         834      833       -1     
  Lines      106589   106465     -124     
==========================================
- Hits        87247    87162      -85     
+ Misses      19342    19303      -39     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@nathanielsimard nathanielsimard merged commit 3dc4b43 into main Nov 29, 2024
11 checks passed
@nathanielsimard nathanielsimard deleted the matmulupdate branch November 29, 2024 20:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants