Skip to content

Commit 9557a50

Browse files
bharrisaubowang007
authored andcommitted
Get windows build working (#1711)
1 parent 1ec8153 commit 9557a50

File tree

4 files changed

+10
-2
lines changed

4 files changed

+10
-2
lines changed

core/plugins/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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
)

core/runtime/CMakeLists.txt

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff 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
4046
install(FILES ${HEADER_FILES} DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/torch_tensorrt/core/runtime")

core/runtime/TRTEngine.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

core/runtime/TRTEngineProfiler.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#include <algorithm>
22
#include <fstream>
33
#include <iomanip>
4+
#include <sstream>
45

56
#include "core/runtime/TRTEngineProfiler.h"
67

0 commit comments

Comments
 (0)