-
Notifications
You must be signed in to change notification settings - Fork 68
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[DOR-614] Move redistributable libs out to a separate file
- Loading branch information
1 parent
53b3061
commit 7dd47c2
Showing
2 changed files
with
100 additions
and
98 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,99 @@ | ||
| if(CMAKE_SYSTEM_NAME STREQUAL "Linux") | ||
| # CUDA toolkit DLLs we depend on: | ||
| set(VERSIONED_CUDA_LIBS | ||
| libcublas*.so* | ||
| libcudart*.so* | ||
| libnvrtc*.so* | ||
| libnvToolsExt*.so* | ||
| ) | ||
|
|
||
| if(CMAKE_SYSTEM_PROCESSOR STREQUAL "aarch64") | ||
| list(APPEND VERSIONED_CUDA_LIBS | ||
| "*cusparse.so*" | ||
| "*curand.so*" | ||
| "*cusolver.so*" | ||
| "*cufft.so*" | ||
| "*cupti.so*" | ||
| ) | ||
| endif() | ||
|
|
||
| foreach(LIB IN LISTS VERSIONED_CUDA_LIBS) | ||
| # torch may bundle it's own specific copy of the cuda libs. if it does, we want everything to point at them | ||
| file(GLOB TORCH_CUDA_LIBS "${TORCH_LIB}/lib/${LIB}") | ||
| if(TORCH_CUDA_LIBS) | ||
| # Sort the list so that we process in order: libX.so -> libX.so.1 -> libX.so.1.1.1 | ||
| list(SORT TORCH_CUDA_LIBS) | ||
| foreach(TORCH_CUDA_LIB IN LISTS TORCH_CUDA_LIBS) | ||
| # create links to the torch bundled libs with hashes in the name | ||
| # e.g. libcublas.so.11 => libcublas-3b81d170.so.11 | ||
| set(target ${TORCH_CUDA_LIB}) | ||
| string(REGEX REPLACE "-[0-9a-f]+[.]" "." link ${target}) | ||
| get_filename_component(target_name ${target} NAME) | ||
| get_filename_component(link_name ${link} NAME) | ||
| if (NOT target_name STREQUAL link_name) | ||
| execute_process(COMMAND ln -rfs ${target_name} ${link_name} WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}) | ||
| install(FILES ${CMAKE_CURRENT_BINARY_DIR}/${link_name} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| # create links to the versioned links above | ||
| # e.g. libcublas.so => libcublas.so.11 | ||
| string(REGEX REPLACE "[.]so[.0-9]*$" ".so" base_link ${link_name}) | ||
| if (NOT base_link STREQUAL link_name) | ||
| execute_process(COMMAND ln -rfs ${link_name} ${base_link} WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}) | ||
| install(FILES ${CMAKE_CURRENT_BINARY_DIR}/${base_link} DESTINATION lib COMPONENT redist_libs) | ||
| endif() | ||
| endif() | ||
| endforeach() | ||
| else() | ||
| # bundle the libraries from the cuda toolkit | ||
| file(GLOB NATIVE_CUDA_LIBS "${CUDAToolkit_TARGET_DIR}/targets/${CMAKE_SYSTEM_PROCESSOR}-linux/lib/${LIB}") | ||
| install(FILES ${NATIVE_CUDA_LIBS} DESTINATION lib COMPONENT redist_libs) | ||
| endif() | ||
| endforeach() | ||
|
|
||
| file(GLOB TORCH_DLLS "${TORCH_LIB}/lib/*.so*") | ||
| install(FILES ${TORCH_DLLS} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| if(DYNAMIC_HDF) | ||
| string(REPLACE "." "" SHARED_LIB_EXT "${CMAKE_SHARED_LIBRARY_SUFFIX}") | ||
| FILTER_LIST("${HDF5_C_LIBRARIES}" DEBUG_LIBRARIES debug optimized ${SHARED_LIB_EXT}) | ||
| RESOLVE_SYMLINKS("${DEBUG_LIBRARIES}" NEW_HDF_DEBUG_LIBRARIES) | ||
| foreach(HDF_LIB IN LISTS NEW_HDF_DEBUG_LIBRARIES) | ||
| if(${HDF_LIB} MATCHES "hdf5") | ||
| install(FILES ${HDF_LIB} DESTINATION lib COMPONENT redist_libs CONFIGURATIONS Debug) | ||
| endif() | ||
| endforeach() | ||
| FILTER_LIST("${HDF5_C_LIBRARIES}" RELEASE_LIBRARIES optimized debug ${SHARED_LIB_EXT}) | ||
| RESOLVE_SYMLINKS("${RELEASE_LIBRARIES}" NEW_HDF_RELEASE_LIBRARIES) | ||
| foreach(HDF_LIB IN LISTS NEW_HDF_RELEASE_LIBRARIES) | ||
| if(${HDF_LIB} MATCHES "hdf5") | ||
| install(FILES ${HDF_LIB} DESTINATION lib COMPONENT redist_libs CONFIGURATIONS Release ReleaseWithDebInfo) | ||
| endif() | ||
| endforeach() | ||
| endif() | ||
|
|
||
| find_library(SZ_DLL sz REQUIRED) | ||
| get_filename_component(SZ_DLL_PATH ${SZ_DLL} DIRECTORY) | ||
| file(GLOB SZ_DLLS "${SZ_DLL_PATH}/libsz.so*") | ||
| install(FILES ${SZ_DLLS} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| find_library(AEC_DLL aec REQUIRED) | ||
| get_filename_component(AEC_DLL_PATH ${AEC_DLL} DIRECTORY) | ||
| file(GLOB AEC_DLLS "${AEC_DLL_PATH}/libaec.so*") | ||
| install(FILES ${AEC_DLLS} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| # If zstd has been dynamically linked, add the .so to the package | ||
| get_filename_component(ZSTD_LIBRARY_PATH ${ZSTD_LIBRARY_RELEASE} DIRECTORY) | ||
| file(GLOB ZSTD_DLLS "${ZSTD_LIBRARY_PATH}/*zstd.so*") | ||
| install(FILES ${ZSTD_DLLS} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| elseif(WIN32) | ||
| file(GLOB TORCH_DLLS "${TORCH_LIB}/lib/*.dll") | ||
| install(FILES ${TORCH_DLLS} DESTINATION bin COMPONENT redist_libs) | ||
| file(GLOB HTSLIB_DLLS "${HTSLIB_DIR}/*.dll") | ||
| install(FILES ${HTSLIB_DLLS} DESTINATION bin COMPONENT redist_libs) | ||
|
|
||
| elseif(APPLE AND NOT CMAKE_SYSTEM_NAME STREQUAL "iOS") | ||
| file(GLOB TORCH_DLLS "${TORCH_LIB}/lib/*.dylib") | ||
| install(FILES ${TORCH_DLLS} DESTINATION lib COMPONENT redist_libs) | ||
|
|
||
| endif() |