coreml : wrap inference call in @autoreleasepool to fix memory leak (g…

…gerganov#1218)
landtanin · Aug 29, 2023 · 89c66bd · 89c66bd
1 parent 87b88a4
commit 89c66bd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/coreml/whisper-encoder.mm b/coreml/whisper-encoder.mm
@@ -53,9 +53,11 @@ void whisper_coreml_encode(
                                            error: nil
     ];
 
-    whisper_encoder_implOutput * outCoreML = [(__bridge id) ctx->data predictionFromLogmel_data:inMultiArray error:nil];
+    @autoreleasepool {
+        whisper_encoder_implOutput * outCoreML = [(__bridge id) ctx->data predictionFromLogmel_data:inMultiArray error:nil];
 
-    memcpy(out, outCoreML.output.dataPointer, outCoreML.output.count * sizeof(float));
+        memcpy(out, outCoreML.output.dataPointer, outCoreML.output.count * sizeof(float));
+    }
 }
 
 #if __cplusplus