From b0b4741a1d81d48c3c428294ed6e943e8cd283b8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=ED=95=9C=EC=A2=85/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Thu, 26 Apr 2018 13:15:44 +0900 Subject: [PATCH] nnapi_test check arguments (#902) This commit fixes when .tflite file is not given from the argument. Which has caused segmentation fault. Signed-off-by: Hanjoung Lee --- tools/nnapi_test/src/nnapi_test.cc | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tools/nnapi_test/src/nnapi_test.cc b/tools/nnapi_test/src/nnapi_test.cc index 4d8d073..137b552 100644 --- a/tools/nnapi_test/src/nnapi_test.cc +++ b/tools/nnapi_test/src/nnapi_test.cc @@ -6,6 +6,8 @@ #include "support/tflite/interp/FlatBufferBuilder.h" #include "support/tflite/Diff.h" +#include + using namespace tflite; using namespace tflite::ops::builtin; @@ -17,6 +19,13 @@ int main(const int argc, char **argv) nnfw::util::env::IntAccessor("VERBOSE").access(verbose); nnfw::util::env::IntAccessor("TOLERANCE").access(tolerance); + if (argc < 2) + { + std::cerr << "nnapi_test\n\n"; + std::cerr << "Usage: " << argv[0] << " <.tflite>\n\n"; + return 1; + } + const auto filename = argv[1]; StderrReporter error_reporter; -- 2.7.4