From e6a6ad697412d80490f6677812c1364462872383 Mon Sep 17 00:00:00 2001 From: Kwanghoon Son Date: Mon, 15 Aug 2022 21:45:51 -0400 Subject: [PATCH] Add snpe lite2_fp32 model test Change-Id: Id9f25f655cc85de348195fe83f54277a3d2abf9b Signed-off-by: Kwanghoon Son --- test/src/inference_engine_tc.cpp | 5 ++++- test/src/inference_engine_test_common.cpp | 3 ++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/test/src/inference_engine_tc.cpp b/test/src/inference_engine_tc.cpp index cb6dfc4..c065433 100644 --- a/test/src/inference_engine_tc.cpp +++ b/test/src/inference_engine_tc.cpp @@ -673,7 +673,10 @@ INSTANTIATE_TEST_CASE_P( // ML Single API for NNStreamer. ParamType_Three( "one", INFERENCE_TARGET_CPU, - { "/usr/share/capi-media-vision/models/IC_Q/tflite/ic_tflite_q_model.tflite" }) + { "/usr/share/capi-media-vision/models/IC_Q/tflite/ic_tflite_q_model.tflite" }), + ParamType_Three( + "snpe", INFERENCE_TARGET_CUSTOM, + { "/usr/share/capi-media-vision/models/OD/snpe/lite2_fp32.dlc" }) /* TODO */ )); diff --git a/test/src/inference_engine_test_common.cpp b/test/src/inference_engine_test_common.cpp index 67036d2..206b983 100644 --- a/test/src/inference_engine_test_common.cpp +++ b/test/src/inference_engine_test_common.cpp @@ -38,7 +38,7 @@ static std::map Model_Formats = { { "caffemodel", INFERENCE_MODEL_CAFFE }, { "pb", INFERENCE_MODEL_TF }, { "tflite", INFERENCE_MODEL_TFLITE }, { "t7", INFERENCE_MODEL_TORCH }, { "weights", INFERENCE_MODEL_DARKNET }, { "xml", INFERENCE_MODEL_DLDT }, - { "onnx", INFERENCE_MODEL_ONNX } + { "onnx", INFERENCE_MODEL_ONNX }, { "dlc", INFERENCE_MODEL_SNPE } }; static MachineCapacity MachineCap[] = { @@ -126,6 +126,7 @@ int GetModelInfo(std::vector &model_paths, break; case INFERENCE_MODEL_TFLITE: case INFERENCE_MODEL_TORCH: + case INFERENCE_MODEL_SNPE: models.push_back(model_paths[0]); break; default: -- 2.34.1