Add library-only build target (#659)

Signed-off-by: Kevin Chen <[email protected]>

Author: kevinch-nv

CMakeLists.txt

@@ -74,14 +74,16 @@ if(BUILD_ONNXIFI)
   set(ONNXIFI_SOURCES onnx_trt_backend.cpp)
 endif()
 
-set(EXECUTABLE_SOURCES
-  main.cpp
-)
-
-set(API_TESTS_SOURCES
-  getSupportedAPITest.cpp
-  ModelImporter.cpp
-)
+# Build executables if BUILD_LIBRARY_ONLY flag is not set
+if (NOT DEFINED BUILD_LIBRARY_ONLY)
+  set(EXECUTABLE_SOURCES
+    main.cpp
+  )
+  set(API_TESTS_SOURCES
+    getSupportedAPITest.cpp
+    ModelImporter.cpp
+  )
+endif()
 
 set(HEADERS
   NvOnnxParser.h
@@ -158,25 +160,27 @@ endif()
 # --------------------------------
 # Converter executable
 # --------------------------------
-add_executable(onnx2trt ${EXECUTABLE_SOURCES})
-target_include_directories(onnx2trt PUBLIC ${ONNX_INCLUDE_DIRS})
-target_link_libraries(onnx2trt PUBLIC ${PROTOBUF_LIB} onnx nvonnxparser_static ${CMAKE_THREAD_LIBS_INIT} ${CMAKE_DL_LIBS}) #${CUDA_LIBRARIES} 
+if (NOT DEFINED BUILD_LIBRARY_ONLY)
+  add_executable(onnx2trt ${EXECUTABLE_SOURCES})
+  target_include_directories(onnx2trt PUBLIC ${ONNX_INCLUDE_DIRS})
+  target_link_libraries(onnx2trt PUBLIC ${PROTOBUF_LIB} onnx nvonnxparser_static ${CMAKE_THREAD_LIBS_INIT} ${CMAKE_DL_LIBS}) #${CUDA_LIBRARIES}
+endif()
 
 # --------------------------------
 # API Tests
 # --------------------------------
-add_executable(getSupportedAPITest ${API_TESTS_SOURCES})
-target_include_directories(getSupportedAPITest PUBLIC ${ONNX_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIR})
-target_link_libraries(getSupportedAPITest PUBLIC ${PROTOBUF_LIB} nvonnxparser_static ${CMAKE_THREAD_LIBS_INIT} ${CMAKE_DL_LIBS}) #${CUDA_LIBRARIES} 
+if (NOT DEFINED BUILD_LIBRARY_ONLY)
+  add_executable(getSupportedAPITest ${API_TESTS_SOURCES})
+  target_include_directories(getSupportedAPITest PUBLIC ${ONNX_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIR})
+  target_link_libraries(getSupportedAPITest PUBLIC ${PROTOBUF_LIB} nvonnxparser_static ${CMAKE_THREAD_LIBS_INIT} ${CMAKE_DL_LIBS}) #${CUDA_LIBRARIES} 
+endif()
 
 # --------------------------------
 # Installation
 # --------------------------------
 install(TARGETS
-                onnx2trt
                 nvonnxparser
                 nvonnxparser_static
-        RUNTIME DESTINATION bin
         LIBRARY DESTINATION lib
         ARCHIVE DESTINATION lib
 )
@@ -185,6 +189,13 @@ install(FILES ${HEADERS}
   DESTINATION include
 )
 
+if (NOT DEFINED BUILD_LIBRARY_ONLY)
+  install(TARGETS
+          onnx2trt
+          RUNTIME DESTINATION bin
+  )
+endif()
+
 SET(CPACK_GENERATOR "DEB")
 SET(CPACK_DEBIAN_PACKAGE_MAINTAINER "Mike Houston") #required
 SET(CPACK_PACKAGE_NAME "onnx-trt-dev")

README.md

@@ -57,6 +57,8 @@ Once you have cloned the repository, you can build the parser libraries and exec
     // Ensure that you update your LD_LIBRARY_PATH to pick up the location of the newly built library:
     export LD_LIBRARY_PATH=$PWD:$LD_LIBRARY_PATH
 
+For building only the libraries, append `-DBUILD_LIBRARY_ONLY=1` to the CMake build command.
+
 ## Executable Usage
 
 ONNX models can be converted to serialized TensorRT engines using the `onnx2trt` executable: