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 -#include +#include #include #include #include