diff --git a/.gitignore b/.gitignore index 22f7f94..1638dd9 100644 --- a/.gitignore +++ b/.gitignore @@ -1,5 +1,6 @@ *~ build/ +installed/ .vscode/ *.bin *.pyc diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h index c6d2472..c2d2661 100644 --- a/include/tkDNN/DarknetParser.h +++ b/include/tkDNN/DarknetParser.h @@ -1,6 +1,6 @@ #pragma once #include -#include "tkDNN/tkdnn.h" +#include "tkdnn.h" namespace tk { namespace dnn { diff --git a/include/tkDNN/NetworkRT.h b/include/tkDNN/NetworkRT.h index a422134..a0e09e7 100644 --- a/include/tkDNN/NetworkRT.h +++ b/include/tkDNN/NetworkRT.h @@ -7,7 +7,7 @@ #include "Layer.h" #include "NvInfer.h" #include -#include +#include "kernels.h" #include #include #include