mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Fix eigen version / kernels.h include

Browse Source
This commit is contained in:
Adam Cuellar
2022-08-25 13:50:30 -04:00
parent a37c14d71d
commit 40dc3f7c2a
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+1 -1
View File
@@ -102,7 +102,7 @@ target_link_libraries(kernels ${CUDA_CUBLAS_LIBRARIES} ${CUDA_LIBRARIES} ${CUDNN
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# External Libraries # External Libraries
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
find_package(Eigen3 3.4 REQUIRED) find_package(Eigen3 REQUIRED)
message("Eigen DIR: " ${EIGEN3_INCLUDE_DIR}) message("Eigen DIR: " ${EIGEN3_INCLUDE_DIR})
include_directories(${EIGEN3_INCLUDE_DIR}) include_directories(${EIGEN3_INCLUDE_DIR})
include/tkDNN/NetworkRT.h
+1 -1
View File
@@ -7,7 +7,7 @@
#include "Layer.h" #include "Layer.h"
#include "NvInfer.h" #include "NvInfer.h"
#include <memory> #include <memory>
#include <tkDNN/kernels.h> #include <kernels.h>
#include <pluginsRT/ActivationLeakyRT.h> #include <pluginsRT/ActivationLeakyRT.h>
#include <pluginsRT/ActivationLogisticRT.h> #include <pluginsRT/ActivationLogisticRT.h>
#include <pluginsRT/ActivationMishRT.h> #include <pluginsRT/ActivationMishRT.h>