From 5f1a7535af0f5b6bc3ebe36b33826a16f38fbfb9 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Fri, 16 Nov 2018 08:07:05 +0900 Subject: [PATCH] [tflchef] Use safemain (#2305) Now, tflchef and tflchef-file uses the main function that safemain provides. Signed-off-by: Jonghyun Park --- contrib/tflchef/tools/console/CMakeLists.txt | 1 + contrib/tflchef/tools/console/Driver.cpp | 2 +- contrib/tflchef/tools/file/CMakeLists.txt | 1 + contrib/tflchef/tools/file/Driver.cpp | 2 +- 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/contrib/tflchef/tools/console/CMakeLists.txt b/contrib/tflchef/tools/console/CMakeLists.txt index b644edc..d9160c3 100644 --- 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 --- a/contrib/tflchef/tools/console/Driver.cpp +++ b/contrib/tflchef/tools/console/Driver.cpp @@ -22,7 +22,7 @@ #include -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 --- 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 --- a/contrib/tflchef/tools/file/Driver.cpp +++ b/contrib/tflchef/tools/file/Driver.cpp @@ -23,7 +23,7 @@ #include #include -int main(int argc, char **argv) +int entry(int argc, char **argv) { if (argc != 3) { -- 2.7.4