fix a offset computation

sonos · Dec 10, 2024 · 4044d80 · 4044d80
1 parent 38f3689
commit 4044d80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metal/src/kernels/matmul/mod.rs b/metal/src/kernels/matmul/mod.rs
@@ -96,7 +96,7 @@ impl GemmDispatchParams {
                     n,
                     k,
                     transpose_a,
-                    a_offset: a_offset + a_batch_idx * n * k * dt.size_of(),
+                    a_offset: a_offset + a_batch_idx * m * k * dt.size_of(),
                     transpose_b,
                     b_offset,
                     c_offset: c_offset + a_batch_idx * m * n * dt.size_of(),