From 6e91f6fda5d093c4c80907d4a2bbb4b437e25d6c Mon Sep 17 00:00:00 2001 From: Tae-Young Chung Date: Thu, 11 Mar 2021 11:19:37 +0900 Subject: [PATCH] MV_INFERENCE_BACKEND_MLAPI with MV_INFERENCE_TARGET_DEVICE_CUSTOM condition only requires to check model config path Change-Id: Ie73cfc3488360d8a5cde0c2c52f92900410b4f7c Signed-off-by: Tae-Young Chung --- mv_inference/inference/include/Inference.h | 5 +++++ mv_inference/inference/src/mv_inference_open.cpp | 2 ++ 2 files changed, 7 insertions(+) diff --git a/mv_inference/inference/include/Inference.h b/mv_inference/inference/include/Inference.h index 27f8f03..320087b 100644 --- a/mv_inference/inference/include/Inference.h +++ b/mv_inference/inference/include/Inference.h @@ -322,6 +322,11 @@ namespace inference engine_config = config; } + int GetTargetType() + { + return mConfig.mTargetTypes; + } + private: bool mCanRun; /**< The flag indicating ready to run Inference */ InferenceConfig mConfig; diff --git a/mv_inference/inference/src/mv_inference_open.cpp b/mv_inference/inference/src/mv_inference_open.cpp index 57ec8fd..c2011b4 100644 --- a/mv_inference/inference/src/mv_inference_open.cpp +++ b/mv_inference/inference/src/mv_inference_open.cpp @@ -175,6 +175,8 @@ int mv_inference_configure_model_open(mv_inference_h infer, backendType < MV_INFERENCE_BACKEND_MAX) && (backendType != MV_INFERENCE_BACKEND_TFLITE) && (backendType != MV_INFERENCE_BACKEND_ARMNN) && + (backendType == MV_INFERENCE_BACKEND_MLAPI && + (pInfer->GetTargetType() & MV_INFERENCE_TARGET_DEVICE_CUSTOM)) && (backendType != MV_INFERENCE_BACKEND_ONE)) { if (access(modelConfigFilePath, F_OK)) { LOGE("modelConfigFilePath in [%s] ", modelConfigFilePath); -- 2.7.4