From: 오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Date: Wed, 24 Jul 2019 09:26:07 +0000 (+0900)
Subject: Set thread count for tflite_run (#5830)
X-Git-Tag: submit/tizen/20190809.050447~442
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=3d1b6308bc056d4fae1f46063af147cecdab795a;p=platform%2Fcore%2Fml%2Fnnfw.git

Set thread count for tflite_run (#5830)

Set thread count for tflite_run by using THREAD environment variable

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
---

diff --git a/tests/tools/tflite_run/src/tflite_run.cc b/tests/tools/tflite_run/src/tflite_run.cc
index b54f0e1..c067f8b 100644
--- a/tests/tools/tflite_run/src/tflite_run.cc
+++ b/tests/tools/tflite_run/src/tflite_run.cc
@@ -73,7 +73,7 @@ int main(const int argc, char **argv)
 
       TFLITE_ENSURE(builder(&interpreter))
 
-      interpreter->SetNumThreads(1);
+      interpreter->SetNumThreads(nnfw::misc::get_env_int("THREAD", -1));
     };
   }
   catch (const std::exception &e)