From: Inki Dae <inki.dae@samsung.com>
Date: Thu, 5 Dec 2024 06:05:03 +0000 (+0900)
Subject: add custom number of threads support
X-Git-Tag: accepted/tizen/9.0/unified/20250102.114432~1
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=034ac2f092578e59aedf68c57d359f8e9b0c4e1c;p=platform%2Fcore%2Fmultimedia%2Finference-engine-tflite.git

add custom number of threads support

Change-Id: I3d550c5921dba536dc113324c0ec25ce37150539
Signed-off-by: Inki Dae <inki.dae@samsung.com>
---

diff --git a/src/inference_engine_tflite.cpp b/src/inference_engine_tflite.cpp
index 948a1c4..81cdbdb 100644
--- a/src/inference_engine_tflite.cpp
+++ b/src/inference_engine_tflite.cpp
@@ -131,7 +131,12 @@ namespace TFLiteImpl
 			}
 		}
 
-		mInterpreter->SetNumThreads(MV_INFERENCE_TFLITE_MAX_THREAD_NUM);
+		const char *envNumThreads = std::getenv("TFLITE_NUM_THREADS");
+		int numThreads = envNumThreads ? std::atoi(envNumThreads) : MV_INFERENCE_TFLITE_MAX_THREAD_NUM;
+
+		LOGD("numThreads : [%d]", numThreads);
+
+		mInterpreter->SetNumThreads(numThreads);
 		LOGI("mInterpreter->tensors_size() :[%zu]",
 			 mInterpreter->tensors_size());