Merge pull request #852 from Jark5455/main

Temporary Fix for GLOG
LaurentMazare · Apr 27, 2024 · 37d90c3 · 37d90c3
2 parents a353eac + d068b18
commit 37d90c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch-sys/build.rs b/torch-sys/build.rs
@@ -385,6 +385,7 @@ impl SystemInfo {
                     .flag(&format!("-Wl,-rpath={}", self.libtorch_lib_dir.display()))
                     .flag("-std=c++17")
                     .flag(&format!("-D_GLIBCXX_USE_CXX11_ABI={}", self.cxx11_abi))
+                    .flag("-DGLOG_USE_GLOG_EXPORT")
                     .files(&c_files)
                     .compile("tch");
             }
@@ -398,6 +399,7 @@ impl SystemInfo {
                     .warnings(false)
                     .includes(&self.libtorch_include_dirs)
                     .flag("/std:c++17")
+                    .flag("/p:DefineConstants=GLOG_USE_GLOG_EXPORT")
                     .files(&c_files)
                     .compile("tch");
             }