aboutsummarylogtreecommitdiffstats
path: root/torchvision-0_17_1-fix-build.patch
blob: 917d041c747bb7c67d54b1aa8390578dbaf28fd8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
--- setup.py.old	2024-03-20 13:32:00.146891902 +0100
+++ setup.py	2024-03-20 13:32:26.983812475 +0100
@@ -189,6 +189,7 @@
     extension = CppExtension
 
     define_macros = []
+    define_macros += [("GLOG_USE_GLOG_EXPORT", None)]
 
     extra_compile_args = {"cxx": []}
     if (torch.cuda.is_available() and ((CUDA_HOME is not None) or is_rocm_pytorch)) or force_cuda:
@@ -250,6 +251,7 @@
 
     # Image reading extension
     image_macros = []
+    image_macros += [("GLOG_USE_GLOG_EXPORT", None)]
     image_include = [extensions_dir]
     image_library = []
     image_link_flags = []
@@ -449,6 +451,7 @@
                 ],
                 extra_compile_args=["-std=c++17"] if os.name != "nt" else ["/std:c++17", "/MP"],
                 extra_link_args=["-std=c++17" if os.name != "nt" else "/std:c++17"],
+                define_macros=[("GLOG_USE_GLOG_EXPORT", None)],
             )
         )
 
@@ -497,6 +500,7 @@
                     "nppicc",
                 ],
                 extra_compile_args=extra_compile_args,
+                define_macros=[("GLOG_USE_GLOG_EXPORT", None)],
             )
         )
     else: