Skip to content

Commit f010dc6

Browse files
committed
Optimize memory when exporting.
1 parent 2458717 commit f010dc6

25 files changed

+997
-285
lines changed

backends/apple/coreml/runtime/delegate/ETCoreMLModelManager.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,7 @@ - (nullable ETCoreMLAsset *)assetWithIdentifier:(NSString *)identifier {
395395
using namespace inmemoryfs;
396396

397397
auto buffer = MemoryBuffer::make_unowned(const_cast<void *>(data.bytes), data.length);
398-
std::unique_ptr<InMemoryFileSystem> inMemoryFS = inmemoryfs::make(buffer);
398+
std::unique_ptr<InMemoryFileSystem> inMemoryFS = inmemoryfs::make_from_buffer(std::move(buffer));
399399
if (!inMemoryFS) {
400400
ETCoreMLLogErrorAndSetNSError(error,
401401
ETCoreMLErrorCorruptedModel,

backends/apple/coreml/runtime/delegate/backend_delegate.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ MLMultiArrayDataType get_data_type(MultiArray::DataType dataType) {
6363
return MLMultiArrayDataTypeFloat32;
6464
}
6565
case MultiArray::DataType::Double: {
66-
return MLMultiArrayDataTypeFloat64;
66+
return MLMultiArrayDataTypeDouble;
6767
}
6868
case MultiArray::DataType::Int: {
6969
return MLMultiArrayDataTypeInt32;

backends/apple/coreml/runtime/delegate/coreml_backend_delegate.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ ModelLoggingOptions get_logging_options(BackendExecutionContext& context) {
149149
ET_CHECK_OR_RETURN_ERROR(handle != nullptr,
150150
InvalidProgram,
151151
"%s: Failed to init the model.", ETCoreMLStrings.delegateIdentifier.UTF8String);
152-
152+
processed->Free();
153153
return handle;
154154
}
155155

0 commit comments

Comments
 (0)