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}
23
23
TensorRT::nvinfer_plugin
24
24
torch
25
25
core_util
26
+ cuDNN::cuDNN
26
27
PRIVATE
27
28
Threads::Threads
28
29
)
Original file line number Diff line number Diff line change @@ -33,8 +33,14 @@ target_link_libraries(${lib_name}
33
33
TensorRT::nvinfer
34
34
torch
35
35
core_util
36
- stdc++fs
37
36
)
38
37
38
+ if (NOT WIN32 )
39
+ target_link_libraries (${lib_name}
40
+ PUBLIC
41
+ stdc++fs
42
+ )
43
+ endif (NOT WIN32 )
44
+
39
45
# Install
40
46
install (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 {
25
25
std::string name;
26
26
RTDevice device_info;
27
27
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() ;
29
29
30
30
std::unordered_map<uint64_t , uint64_t > in_binding_map = {}; // TRT IDX -> PYT IDX
31
31
std::unordered_map<uint64_t , uint64_t > out_binding_map = {}; // TRT IDX -> PYT IDX
Original file line number Diff line number Diff line change 1
1
#include < algorithm>
2
2
#include < fstream>
3
3
#include < iomanip>
4
+ #include < sstream>
4
5
5
6
#include " core/runtime/TRTEngineProfiler.h"
6
7
You can’t perform that action at this time.
0 commit comments