extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-04-24 11:18:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

coreml : wrap inference call in @autoreleasepool to fix memory leak (#1218)

Browse Source
This commit is contained in:
Dener Stassun 2023-08-29 09:44:38 -03:00 committed by GitHub
parent 8e46ba80d3
commit 18e9889418
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
coreml/whisper-encoder.mm
View File

@ -53,10 +53,12 @@ void whisper_coreml_encode(
error: nil error: nil
]; ];
@autoreleasepool {
whisper_encoder_implOutput * outCoreML = [(__bridge id) ctx->data predictionFromLogmel_data:inMultiArray error:nil]; 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 #if __cplusplus
} }