Add cmake support for the examples

The examples are considered as completely separated cmake project, in
order to demonstrate how torchtrt would typically be consumed in a
different CMake project and test the torchtrtConfig.cmake.

Signed-off-by: Gabriel Cuendet <[email protected]>
Signed-off-by: Gabriel Cuendet <[email protected]>

Author: gcuendet-cognex

CMakeLists.txt

@@ -13,8 +13,8 @@ if (DEFINED CMAKE_MODULE_PATH)
 endif()
 
 include(cmake/build_options.cmake)
-include(cmake/dependencies.cmake)
 include(cmake/paths.cmake)
+include(cmake/dependencies.cmake)
 if(MSVC)
 	add_compile_options(/wd4624 /wd4067 /permissive-)
   # When using Ninja generator, suppress the warning D9025

Config.cmake.in

@@ -2,9 +2,12 @@
 
 include(CMakeFindDependencyMacro)
 
-find_dependency(Torch REQUIRED CONFIG)
-find_dependency(TensorRT REQUIRED)
-
+find_dependency(Torch)
+find_package(TensorRT QUIET)
+if (NOT TensorRT_FOUND)
+    list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_LIST_DIR}/Modules")
+    find_dependency(TensorRT)
+endif()
 include("${CMAKE_CURRENT_LIST_DIR}/torchtrtTargets.cmake")
 
 check_required_components(MathFunctions)

cmake/dependencies.cmake

@@ -1,5 +1,13 @@
 # TensorRT
-find_package(TensorRT REQUIRED)
+find_package(TensorRT QUIET)
+if (NOT TensorRT_FOUND)
+    list(APPEND CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/cmake/Modules")
+    find_package(TensorRT REQUIRED)
+endif()
+
+# If the custom finders are needed at this point, there are good chances that they will be needed when consuming the library as well
+install(FILES "${CMAKE_SOURCE_DIR}/cmake/Modules/FindTensorRT.cmake" DESTINATION "${CMAKE_INSTALL_LIBDIR}/cmake/torchtrt/Modules")
+install(FILES "${CMAKE_SOURCE_DIR}/cmake/Modules/FindcuDNN.cmake" DESTINATION "${CMAKE_INSTALL_LIBDIR}/cmake/torchtrt/Modules")
 
 # CUDA
 find_package(CUDAToolkit REQUIRED)

examples/int8/ptq/CMakeLists.txt

@@ -0,0 +1,17 @@
+cmake_minimum_required(VERSION 3.17)
+project(ptq LANGUAGES CXX)
+
+# Find dependencies
+find_package(Torch REQUIRED)
+find_package(torchtrt REQUIRED)
+
+set(SRCS
+    ${CMAKE_SOURCE_DIR}/main.cpp
+    ${CMAKE_SOURCE_DIR}/../benchmark/benchmark.cpp
+    ${CMAKE_SOURCE_DIR}/../datasets/cifar10.cpp
+
+)
+
+add_executable(${CMAKE_PROJECT_NAME} ${SRCS})
+target_include_directories(${CMAKE_PROJECT_NAME} PRIVATE "${CMAKE_SOURCE_DIR}/../../..")
+target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE torch "-Wl,--no-as-needed" torchtrt "-Wl,--as-needed")

examples/int8/qat/CMakeLists.txt

@@ -0,0 +1,17 @@
+cmake_minimum_required(VERSION 3.17)
+project(qat LANGUAGES CXX)
+
+# Find dependencies
+find_package(Torch REQUIRED)
+find_package(torchtrt REQUIRED)
+
+set(SRCS
+    ${CMAKE_SOURCE_DIR}/main.cpp
+    ${CMAKE_SOURCE_DIR}/../benchmark/benchmark.cpp
+    ${CMAKE_SOURCE_DIR}/../datasets/cifar10.cpp
+
+)
+
+add_executable(${CMAKE_PROJECT_NAME} ${SRCS})
+target_include_directories(${CMAKE_PROJECT_NAME} PRIVATE "${CMAKE_SOURCE_DIR}/../../..")
+target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE torch "-Wl,--no-as-needed" torchtrt "-Wl,--as-needed")

examples/torchtrt_runtime_example/CMakeLists.txt

@@ -0,0 +1,13 @@
+cmake_minimum_required(VERSION 3.17)
+project(torchtrt_runtime_example LANGUAGES CXX)
+
+# Find dependencies
+find_package(Torch REQUIRED)
+find_package(torchtrt REQUIRED)
+
+set(SRCS
+    main.cpp
+)
+
+add_executable(${CMAKE_PROJECT_NAME} ${SRCS})
+target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE torch "-Wl,--no-as-needed" torchtrt_runtime "-Wl,--as-needed")