diff --git a/core/plugins/CMakeLists.txt b/core/plugins/CMakeLists.txt index 90b039c94a..2b1eaaf73f 100644 --- a/core/plugins/CMakeLists.txt +++ b/core/plugins/CMakeLists.txt @@ -23,6 +23,7 @@ target_link_libraries(${lib_name} TensorRT::nvinfer_plugin torch core_util + cuDNN::cuDNN PRIVATE Threads::Threads ) diff --git a/core/runtime/CMakeLists.txt b/core/runtime/CMakeLists.txt index d21c661af1..610dbf46f7 100644 --- a/core/runtime/CMakeLists.txt +++ b/core/runtime/CMakeLists.txt @@ -33,8 +33,14 @@ target_link_libraries(${lib_name} TensorRT::nvinfer torch core_util - stdc++fs ) +if(NOT WIN32) + target_link_libraries(${lib_name} + PUBLIC + stdc++fs + ) +endif(NOT WIN32) + # Install install(FILES ${HEADER_FILES} DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/torch_tensorrt/core/runtime") diff --git a/core/runtime/TRTEngine.h b/core/runtime/TRTEngine.h index 5615a824a2..1a2df1f2e3 100644 --- a/core/runtime/TRTEngine.h +++ b/core/runtime/TRTEngine.h @@ -25,7 +25,7 @@ struct TRTEngine : torch::CustomClassHolder { std::string name; RTDevice device_info; - std::string profile_path_prefix = std::experimental::filesystem::temp_directory_path(); + std::string profile_path_prefix = std::experimental::filesystem::temp_directory_path().string(); std::unordered_map in_binding_map = {}; // TRT IDX -> PYT IDX std::unordered_map out_binding_map = {}; // TRT IDX -> PYT IDX diff --git a/core/runtime/TRTEngineProfiler.cpp b/core/runtime/TRTEngineProfiler.cpp index a115951610..8f7f0ac4e9 100644 --- a/core/runtime/TRTEngineProfiler.cpp +++ b/core/runtime/TRTEngineProfiler.cpp @@ -1,6 +1,7 @@ #include #include #include +#include #include "core/runtime/TRTEngineProfiler.h"