Statically linking zlib and libpng

Author: r-zenine

setup.py

@@ -93,13 +93,13 @@ def get_extensions():
     extra_compile_args = {}
     third_party_search_directories = []
 
-
-    runtime_library_dirs = None
+    extra_objects = []
     if sys.platform.startswith('linux'):
         sources = sources + source_image_cpu
         libraries.append('png')
         third_party_search_directories.append(os.path.join(cwd, "third_party/libpng"))
         runtime_library_dirs = ['lib']
+        extra_objects = ['third_party/zlib/libz.a', 'third_party/libpng/libpng.a']
 
     extension = CppExtension
 
@@ -117,7 +117,6 @@ def get_extensions():
 
     define_macros = []
 
-    extra_compile_args = {}
     if (torch.cuda.is_available() and CUDA_HOME is not None) or os.getenv('FORCE_CUDA', '0') == '1':
         extension = CUDAExtension
         sources += source_cuda
@@ -162,7 +161,7 @@ def get_extensions():
             include_dirs=include_dirs + third_party_search_directories,
             define_macros=define_macros,
             extra_compile_args=extra_compile_args,
-            runtime_library_dirs=runtime_library_dirs
+            extra_objects=extra_objects
         )
     ]
     if compile_cpp_tests:
@@ -231,9 +230,9 @@ def build_deps():
         os.chdir(this_dir)
 
         zlib_path = os.path.join(this_dir, "third_party/zlib")
-        libpng_cmake_options = "-DPNG_BUILD_ZLIB=ON -DPNG_STATIC=OFF -DZLIB_INCLUDE_DIR:PATH={zlib_path} -DZLIB_LIBRARY:FILEPATH={zlib_path}/libz.so".format(zlib_path=zlib_path)
+        libpng_cmake_options = "-DPNG_BUILD_ZLIB=ON -DPNG_STATIC=ON -DZLIB_INCLUDE_DIR:PATH={zlib_path} -DZLIB_LIBRARY:FILEPATH={zlib_path}/libz.so".format(zlib_path=zlib_path)
         os.chdir("third_party/libpng/")
-        os.system('cmake {} .'.format(libpng_cmake_options))
+        os.system('cmake -DCMAKE_C_FLAGS="-fPIC" {} .'.format(libpng_cmake_options))
         throw_of_failure("cmake --build . -- -j {}".format(cpu_count))
         os.chdir(this_dir)
 
@@ -243,15 +242,6 @@ def build_ext_with_dependencies(self):
     return BuildExtension.with_options(no_python_abi_suffix=True)(self)
 
 
-data_files = []
-if sys.platform.startswith('linux'):
-    data_files = [
-        ('torchvision/lib', [
-            'third_party/zlib/libz.so',
-            'third_party/libpng/libpng.so'])
-    ]
-
-
 setup(
     # Metadata
     name=package_name,
@@ -276,5 +266,4 @@ def build_ext_with_dependencies(self):
         'build_ext': build_ext_with_dependencies,
         'clean': clean,
     },
-    data_files=data_files
 )