From d47b6f94a6af62d6f25f80c61477998bec814dfa Mon Sep 17 00:00:00 2001
From: Hubert de La Jonquiere <hubert.delajonquiere@sonos.com>
Date: Fri, 13 Dec 2024 09:52:00 +0100
Subject: [PATCH] Fix MPS command buffer

---
 metal/src/kernels/matmul/mps/api.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/metal/src/kernels/matmul/mps/api.rs b/metal/src/kernels/matmul/mps/api.rs
index 37bec85e3b..dcef69b747 100644
--- a/metal/src/kernels/matmul/mps/api.rs
+++ b/metal/src/kernels/matmul/mps/api.rs
@@ -1,6 +1,6 @@
 use foreign_types::{foreign_type, ForeignType};
 use metal::mps::{Kernel, KernelRef};
-use metal::{Buffer, Device, NSUInteger};
+use metal::{Buffer, CommandBuffer, Device, NSUInteger};
 use objc::runtime::Object;
 use objc::{class, msg_send, sel, sel_impl};
 use paste::paste;
@@ -277,8 +277,9 @@ impl MatrixMultiplicationRef {
         right: Matrix,
         result: Matrix,
     ) {
+        let cmd_buffer: CommandBuffer = command_buffer.as_ref().to_owned();
         unsafe {
-            msg_send![self, encodeToCommandBuffer: command_buffer
+            msg_send![self, encodeToCommandBuffer: cmd_buffer
                                   leftMatrix: left
                                   rightMatrix: right
                                   resultMatrix: result]
@@ -354,8 +355,9 @@ impl MatrixVectorMultiplicationRef {
         right: Vector,
         result: Vector,
     ) {
+        let cmd_buffer: CommandBuffer = command_buffer.as_ref().to_owned();
         unsafe {
-            msg_send![self, encodeToCommandBuffer: command_buffer
+            msg_send![self, encodeToCommandBuffer: cmd_buffer
                                   inputMatrix: left
                                   inputVector: right
                                   resultVector: result]