Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

support CoreML for SetBuiltInDelegate

This commit is contained in:
qinzheng 2022-12-16 16:43:36 +08:00 committed by qinzheng4
parent fca2ba5641
commit b64279dc61
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mindspore/lite/src/litert/lite_session.cc
View File

@ -53,6 +53,9 @@
#if SUPPORT_NNAPI
#include "src/litert/delegate/nnapi/nnapi_delegate.h"
#endif
#ifdef ENABLE_COREML
#include "src/litert/delegate/coreml/coreml_delegate.h"
#endif
#include "src/litert/runtime_convert.h"
#include "extendrt/mindir_loader/model_loader.h"
#ifndef __ANDROID__
@ -950,6 +953,18 @@ int LiteSession::CreateNNAPIDelegate() {
return RET_OK;
}
int LiteSession::CreateCoreMLDelegate() {
#ifdef ENABLE_COREML
delegate_ = std::make_shared<CoreMLDelegate>();
if (delegate_ == nullptr) {
MS_LOG(ERROR) << "New delegate_ failed";
return RET_ERROR;
}
this->context_->delegate = delegate_;
#endif
return RET_OK;
}
int LiteSession::DelegateInit() {
#ifndef DELEGATE_CLIP
int ret = RET_OK;
@ -961,6 +976,9 @@ int LiteSession::DelegateInit() {
case kNNAPI:
ret = CreateNNAPIDelegate();
break;
case kCoreML:
ret = CreateCoreMLDelegate();
break;
default:
MS_LOG(ERROR) << "Unsupported built-in delegate mode: " << context_->delegate_mode_;
return RET_ERROR;