extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-01-01 03:28:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

coreml : fix memory leak (#899)

Browse Source
This commit is contained in:
Georgi Gerganov 2023-05-09 18:38:12 +03:00 committed by GitHub
parent bf2449dfae
commit 1d17cd5bb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
coreml/whisper-encoder.mm
View File

@ -51,15 +51,10 @@ void whisper_coreml_encode(
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];
MLMultiArray * outMA = outCoreML.output; memcpy(out, outCoreML.output.dataPointer, outCoreML.output.count * sizeof(float));
//NSArray<NSNumber *> * shape = outMA.shape; [outCoreML release];
//NSArray<NSNumber *> * strides = outMA.strides; [inMultiArray release];
//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));
} }
#if __cplusplus #if __cplusplus