From f911018186326850d1f3c883e3f0d4a63ec25cbe Mon Sep 17 00:00:00 2001 From: Inki Dae Date: Thu, 11 Jun 2020 16:16:42 +0900 Subject: [PATCH] Set supported_device_types according to MLAPI backend type NNFW supports only CPU and GPU accelerated NN runtime so Consider using NNFW tensor filter plugin of NNStreamer. Change-Id: I3ed4ae5018b984c812f8bad69eebbfdae69dd030 Signed-off-by: Inki Dae --- src/inference_engine_mlapi.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/inference_engine_mlapi.cpp b/src/inference_engine_mlapi.cpp index ab36c9b..d6d4706 100644 --- a/src/inference_engine_mlapi.cpp +++ b/src/inference_engine_mlapi.cpp @@ -366,7 +366,12 @@ namespace MLAPIImpl } // TODO. flag supported accel device types according to a given ML Single API of nnstreamer backend. - capacity->supported_accel_devices = INFERENCE_TARGET_CUSTOM; + if (mPluginType == INFERENCE_BACKEND_MLAPI) { + capacity->supported_accel_devices = INFERENCE_TARGET_CUSTOM; + } else { + capacity->supported_accel_devices = INFERENCE_TARGET_GPU | + INFERENCE_TARGET_CPU; + } LOGI("LEAVE"); -- 2.7.4