mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
c284a686f2
### Description
AppendExecutionProvider("CoreML", {{"MLComputeUnits","MLProgram"}})
### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
---------
Co-authored-by: Scott McKay <skottmckay@gmail.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
63 lines
2.1 KiB
Text
63 lines
2.1 KiB
Text
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#import "ort_coreml_execution_provider.h"
|
|
|
|
#import "cxx_api.h"
|
|
#import "error_utils.h"
|
|
#import "ort_session_internal.h"
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
BOOL ORTIsCoreMLExecutionProviderAvailable() {
|
|
return ORT_OBJC_API_COREML_EP_AVAILABLE ? YES : NO;
|
|
}
|
|
|
|
@implementation ORTCoreMLExecutionProviderOptions
|
|
|
|
@end
|
|
|
|
@implementation ORTSessionOptions (ORTSessionOptionsCoreMLEP)
|
|
|
|
- (BOOL)appendCoreMLExecutionProviderWithOptions:(ORTCoreMLExecutionProviderOptions*)options
|
|
error:(NSError**)error {
|
|
#if ORT_OBJC_API_COREML_EP_AVAILABLE
|
|
try {
|
|
const uint32_t flags =
|
|
(options.useCPUOnly ? COREML_FLAG_USE_CPU_ONLY : 0) |
|
|
(options.useCPUAndGPU ? COREML_FLAG_USE_CPU_AND_GPU : 0) |
|
|
(options.enableOnSubgraphs ? COREML_FLAG_ENABLE_ON_SUBGRAPH : 0) |
|
|
(options.onlyEnableForDevicesWithANE ? COREML_FLAG_ONLY_ENABLE_DEVICE_WITH_ANE : 0) |
|
|
(options.onlyAllowStaticInputShapes ? COREML_FLAG_ONLY_ALLOW_STATIC_INPUT_SHAPES : 0) |
|
|
(options.createMLProgram ? COREML_FLAG_CREATE_MLPROGRAM : 0);
|
|
|
|
Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CoreML(
|
|
[self CXXAPIOrtSessionOptions], flags));
|
|
return YES;
|
|
}
|
|
ORT_OBJC_API_IMPL_CATCH_RETURNING_BOOL(error);
|
|
#else // !ORT_OBJC_API_COREML_EP_AVAILABLE
|
|
static_cast<void>(options);
|
|
ORTSaveCodeAndDescriptionToError(ORT_FAIL, "CoreML execution provider is not enabled.", error);
|
|
return NO;
|
|
#endif
|
|
}
|
|
|
|
- (BOOL)appendCoreMLExecutionProviderWithOptionsV2:(NSDictionary*)provider_options
|
|
error:(NSError**)error {
|
|
#if ORT_OBJC_API_COREML_EP_AVAILABLE
|
|
try {
|
|
return [self appendExecutionProvider:@"CoreML" providerOptions:provider_options error:error];
|
|
}
|
|
ORT_OBJC_API_IMPL_CATCH_RETURNING_BOOL(error);
|
|
|
|
#else // !ORT_OBJC_API_COREML_EP_AVAILABLE
|
|
static_cast<void>(provider_options);
|
|
ORTSaveCodeAndDescriptionToError(ORT_FAIL, "CoreML execution provider is not enabled.", error);
|
|
return NO;
|
|
#endif
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|