From 1d17cd5bb37a3212679d6055ad69ba5a8d58eb71 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Tue, 9 May 2023 18:38:12 +0300
Subject: [PATCH] coreml : fix memory leak (#899)

---
 coreml/whisper-encoder.mm | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/coreml/whisper-encoder.mm b/coreml/whisper-encoder.mm
index dd08f0f3..49aace8c 100644
--- a/coreml/whisper-encoder.mm
+++ b/coreml/whisper-encoder.mm
@@ -51,15 +51,10 @@ void whisper_coreml_encode(
 
     whisper_encoder_implOutput * outCoreML = [(__bridge id) ctx->data predictionFromLogmel_data:inMultiArray error:nil];
 
-    MLMultiArray * outMA = outCoreML.output;
+    memcpy(out, outCoreML.output.dataPointer, outCoreML.output.count * sizeof(float));
 
-    //NSArray<NSNumber *> * shape = outMA.shape;
-    //NSArray<NSNumber *> * strides = outMA.strides;
-
-    //printf("shape:   %ld %ld %ld %ld\n", [shape[0] longValue], [shape[1] longValue], [shape[2] longValue], [shape[3] longValue]);
-    //printf("strides: %ld %ld %ld %ld\n", [strides[0] longValue], [strides[1] longValue], [strides[2] longValue], [strides[3] longValue]);
-
-    memcpy(out, outMA.dataPointer, outMA.count * sizeof(float));
+    [outCoreML release];
+    [inMultiArray release];
 }
 
 #if __cplusplus