diff --git a/torch-sys/build.rs b/torch-sys/build.rs index 637bb50e..82fc6240 100644 --- 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"); }