ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-12-19 04:37:51 +00:00
Code Issues Actions 4 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
];
@autoreleasepool {
whisper_encoder_implOutput * outCoreML = [(__bridge id) ctx->data predictionFromLogmel_data:inMultiArray error:nil];
memcpy(out, outCoreML.output.dataPointer, outCoreML.output.count * sizeof(float));
}
}
#if __cplusplus
}