Merge pull request #103120 from danieldk/libtorch-bin-cmake-fix-lib-paths

libtorch-bin: fix CMake library paths for split outputs
This commit is contained in:
Daniël de Kok 2020-11-10 12:10:38 +01:00 committed by GitHub
commit da115b4c56
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions

View file

@ -45,6 +45,14 @@ in stdenv.mkDerivation {
# We do not care about Java support...
rm -f $out/lib/lib*jni* 2> /dev/null || true
# Fix up library paths for split outputs
substituteInPlace $dev/share/cmake/Torch/TorchConfig.cmake \
--replace \''${TORCH_INSTALL_PREFIX}/lib "$out/lib" \
substituteInPlace \
$dev/share/cmake/Caffe2/Caffe2Targets-release.cmake \
--replace \''${_IMPORT_PREFIX}/lib "$out/lib" \
'';
postFixup = let

View file

@ -1,3 +1,4 @@
cmake_minimum_required(VERSION 3.0)
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
add_executable(test test.cpp)