From 4044d80d0e02aedd76c6674dd5009121a7a72d17 Mon Sep 17 00:00:00 2001
From: Mathieu Poumeyrol <mathieu.poumeyrol@sonos.com>
Date: Tue, 10 Dec 2024 10:47:23 +0100
Subject: [PATCH] fix a offset computation

---
 metal/src/kernels/matmul/mod.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metal/src/kernels/matmul/mod.rs b/metal/src/kernels/matmul/mod.rs
index be85935bd4..2845e5e6ce 100644
--- 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(),