File tree Expand file tree Collapse file tree 4 files changed +10
-2
lines changed Expand file tree Collapse file tree 4 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -23,6 +23,7 @@ target_link_libraries(${lib_name}
2323 TensorRT::nvinfer_plugin
2424 torch
2525 core_util
26+ cuDNN::cuDNN
2627 PRIVATE
2728 Threads::Threads
2829)
Original file line number Diff line number Diff line change @@ -33,8 +33,14 @@ target_link_libraries(${lib_name}
3333 TensorRT::nvinfer
3434 torch
3535 core_util
36- stdc++fs
3736)
3837
38+ if (NOT WIN32 )
39+ target_link_libraries (${lib_name}
40+ PUBLIC
41+ stdc++fs
42+ )
43+ endif (NOT WIN32 )
44+
3945# Install
4046install (FILES ${HEADER_FILES} DESTINATION "${CMAKE_INSTALL_INCLUDEDIR} /torch_tensorrt/core/runtime" )
Original file line number Diff line number Diff line change @@ -25,7 +25,7 @@ struct TRTEngine : torch::CustomClassHolder {
2525 std::string name;
2626 RTDevice device_info;
2727
28- std::string profile_path_prefix = std::experimental::filesystem::temp_directory_path();
28+ std::string profile_path_prefix = std::experimental::filesystem::temp_directory_path().string() ;
2929
3030 std::unordered_map<uint64_t , uint64_t > in_binding_map = {}; // TRT IDX -> PYT IDX
3131 std::unordered_map<uint64_t , uint64_t > out_binding_map = {}; // TRT IDX -> PYT IDX
Original file line number Diff line number Diff line change 11#include < algorithm>
22#include < fstream>
33#include < iomanip>
4+ #include < sstream>
45
56#include " core/runtime/TRTEngineProfiler.h"
67
You can’t perform that action at this time.
0 commit comments