From 40dc3f7c2a855e83d18cb7da5d21ed208ec3ac34 Mon Sep 17 00:00:00 2001
From: Adam Cuellar <cuellar.adam@gmail.com>
Date: Thu, 25 Aug 2022 13:50:30 -0400
Subject: [PATCH] Fix eigen version / kernels.h include

---
 CMakeLists.txt            | 2 +-
 include/tkDNN/NetworkRT.h | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 75c785b..c29e987 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -102,7 +102,7 @@ target_link_libraries(kernels ${CUDA_CUBLAS_LIBRARIES} ${CUDA_LIBRARIES} ${CUDNN
 #-------------------------------------------------------------------------------
 # External Libraries
 #-------------------------------------------------------------------------------
-find_package(Eigen3 3.4 REQUIRED)
+find_package(Eigen3 REQUIRED)
 message("Eigen DIR: " ${EIGEN3_INCLUDE_DIR})
 include_directories(${EIGEN3_INCLUDE_DIR})
 
diff --git a/include/tkDNN/NetworkRT.h b/include/tkDNN/NetworkRT.h
index b607644..a8138a1 100644
--- a/include/tkDNN/NetworkRT.h
+++ b/include/tkDNN/NetworkRT.h
@@ -7,7 +7,7 @@
 #include "Layer.h"
 #include "NvInfer.h"
 #include <memory>
-#include <tkDNN/kernels.h>
+#include <kernels.h>
 #include <pluginsRT/ActivationLeakyRT.h>
 #include <pluginsRT/ActivationLogisticRT.h>
 #include <pluginsRT/ActivationMishRT.h>