Get windows build working (#1711)

pytorch · Mar 21, 2023 · 93b0400 · 93b0400
1 parent a1d4af0
commit 93b0400
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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<uint64_t, uint64_t> in_binding_map = {}; // TRT IDX -> PYT IDX
   std::unordered_map<uint64_t, uint64_t> out_binding_map = {}; // TRT IDX -> PYT IDX

diff --git a/core/runtime/TRTEngineProfiler.cpp b/core/runtime/TRTEngineProfiler.cpp
@@ -1,6 +1,7 @@
 #include <algorithm>
 #include <fstream>
 #include <iomanip>
+#include <sstream>
 
 #include "core/runtime/TRTEngineProfiler.h"