[tflchef] Use safemain (#2305)

Now, tflchef and tflchef-file uses the main function that safemain
provides.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/tflchef/tools/console/CMakeLists.txt b/contrib/tflchef/tools/console/CMakeLists.txt
index b644edc..d9160c3 100644 (file)
--- a/contrib/tflchef/tools/console/CMakeLists.txt
+++ b/contrib/tflchef/tools/console/CMakeLists.txt
@@ -1,2 +1,3 @@
 add_executable(tflchef Driver.cpp)
 target_link_libraries(tflchef tflchef_core)
+target_link_libraries(tflchef safemain)

diff --git a/contrib/tflchef/tools/console/Driver.cpp b/contrib/tflchef/tools/console/Driver.cpp
index 186c184..0f319c0 100644 (file)
--- a/contrib/tflchef/tools/console/Driver.cpp
+++ b/contrib/tflchef/tools/console/Driver.cpp
@@ -22,7 +22,7 @@
 
 #include <iostream>
 
-int main(int argc, char **argv)
+int entry(int argc, char **argv)
 {
   ::tflchef::ModelRecipe model_recipe;
 

diff --git a/contrib/tflchef/tools/file/CMakeLists.txt b/contrib/tflchef/tools/file/CMakeLists.txt
index 3f6e3a8..477b7d9 100644 (file)
--- a/contrib/tflchef/tools/file/CMakeLists.txt
+++ b/contrib/tflchef/tools/file/CMakeLists.txt
@@ -1,2 +1,3 @@
 add_executable(tflchef-file Driver.cpp)
 target_link_libraries(tflchef-file tflchef_core)
+target_link_libraries(tflchef-file safemain)

diff --git a/contrib/tflchef/tools/file/Driver.cpp b/contrib/tflchef/tools/file/Driver.cpp
index d8295c8..1287b01 100644 (file)
--- a/contrib/tflchef/tools/file/Driver.cpp
+++ b/contrib/tflchef/tools/file/Driver.cpp
@@ -23,7 +23,7 @@
 #include <fstream>
 #include <iostream>
 
-int main(int argc, char **argv)
+int entry(int argc, char **argv)
 {
   if (argc != 3)
   {