From 443179359d42ad2cc45c120f360c3cdf299fe2c3 Mon Sep 17 00:00:00 2001
From: Francesco Gatti <gattifrancesco@hotmail.it>
Date: Mon, 3 Dec 2018 22:04:04 +0100
Subject: [PATCH] config

---
 CMakeLists.txt    |  3 +++
 tkDNNConfig.cmake | 10 ++++++++++
 2 files changed, 13 insertions(+)
 create mode 100644 tkDNNConfig.cmake

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 6454f1b..6750f19 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -95,3 +95,6 @@ message("install dir:" ${CMAKE_INSTALL_PREFIX})
 install(DIRECTORY include/ DESTINATION include/${CMAKE_PROJECT_NAME}
 FILES_MATCHING PATTERN "*.h")
 install(TARGETS tkDNN kernels DESTINATION lib)
+install(FILES "${CMAKE_SOURCE_DIR}/${CMAKE_PROJECT_NAME}Config.cmake" # source directory
+        DESTINATION "share/${CMAKE_PROJECT_NAME}/cmake/" # target directory
+)
diff --git a/tkDNNConfig.cmake b/tkDNNConfig.cmake
new file mode 100644
index 0000000..8665415
--- /dev/null
+++ b/tkDNNConfig.cmake
@@ -0,0 +1,10 @@
+find_package(CUDA REQUIRED)
+find_package(OpenCV REQUIRED)
+find_library(NVINFER NAMES nvinfer)
+if(NVINFER STREQUAL "NVINFER-NOTFOUND")
+    set(NVINFER_INCLUDES "/usr/local/nvidia/tensorrt/include/")
+    link_directories(/usr/local/nvidia/tensorrt/targets/x86_64-linux-gnu/lib/ 
+		     /usr/local/cuda/targets/x86_64-linux/lib/)
+endif()
+set(tkDNN_INCLUDE_DIRS ${CUDA_INCLUDE_DIRS} ${OPENCV_INCLUDE_DIRS} ${NVINFER_INCLUDES})
+set(tkDNN_LIBRARIES tkDNN kernels ${CUDA_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES} -lcudnn -lnvinfer ${OpenCV_LIBS})