futurile/guix-play
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

gnu: liblantern: Do not hardcode Python version.

Browse Source 
* gnu/packages/machine-learning.scm (liblantern)[arguments]: Do not hardcode
Python version.
This commit is contained in:
Ricardo Wurmus 2022-09-19 20:26:00 +02:00
parent 851441a5d9
commit 7531efe54d
No known key found for this signature in database
GPG Key ID: 197A5888235FACAC
1 changed files with 41 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
gnu/packages/machine-learning.scm
View File

@ -3180,42 +3180,47 @@ of Hidden Markov Models.")
(list
#:tests? #false ;no test target
#:phases
#~(modify-phases %standard-phases
(add-after 'unpack 'chdir
(lambda _ (chdir "lantern")))
(add-after 'chdir 'do-not-download-binaries
(lambda* (#:key inputs #:allow-other-keys)
(substitute* "CMakeLists.txt"
(("find_package\\(Torch.*") "set(TORCH_CXX_FLAGS \"-ltorch\")\n")
(("retrieve_lib\\(.*") ""))
(setenv "LIBRARY_PATH"
(string-append
(search-input-directory
inputs "/lib/python3.9/site-packages/torch/lib")
":" (or (getenv "LIBRARY_PATH") "")))
(setenv "CPLUS_INCLUDE_PATH"
(string-append
(search-input-directory
inputs "lib/python3.9/site-packages/torch/include/torch/csrc/api/include/")
":"
(search-input-directory
inputs "lib/python3.9/site-packages/torch/include/")
":"
(or (getenv "CPLUS_INCLUDE_PATH") "")))
(setenv "C_INCLUDE_PATH"
(string-append
(search-input-directory
inputs "lib/python3.9/site-packages/torch/include/")
":"
(or (getenv "C_INCLUDE_PATH") "")))))
(replace 'install
(lambda _
(install-file
"../build/liblantern.so"
(string-append #$output "/lib"))
(copy-recursively
"../lantern/include"
(string-append #$output "/include")))))))
(let ((python-version (version-major+minor (package-version python))))
#~(modify-phases %standard-phases
(add-after 'unpack 'chdir
(lambda _ (chdir "lantern")))
(add-after 'chdir 'do-not-download-binaries
(lambda* (#:key inputs #:allow-other-keys)
(substitute* "CMakeLists.txt"
(("find_package\\(Torch.*") "set(TORCH_CXX_FLAGS \"-ltorch\")\n")
(("retrieve_lib\\(.*") ""))
(let ((site-packages (string-append "/lib/python"
#$python-version
"/site-packages")))
(setenv "LIBRARY_PATH"
(string-append
(search-input-directory
inputs (string-append site-packages "/torch/lib"))
":" (or (getenv "LIBRARY_PATH") "")))
(setenv "CPLUS_INCLUDE_PATH"
(string-append
(search-input-directory
inputs (string-append
site-packages "/torch/include/torch/csrc/api/include/"))
":"
(search-input-directory
inputs (string-append site-packages "/torch/include/"))
":"
(or (getenv "CPLUS_INCLUDE_PATH") "")))
(setenv "C_INCLUDE_PATH"
(string-append
(search-input-directory
inputs (string-append site-packages "/torch/include/"))
":"
(or (getenv "C_INCLUDE_PATH") ""))))))
(replace 'install
(lambda _
(install-file
"../build/liblantern.so"
(string-append #$output "/lib"))
(copy-recursively
"../lantern/include"
(string-append #$output "/include"))))))))
(inputs (list python-pytorch-for-r-torch))
(home-page "https://github.com/mlverse/torch/")
(synopsis "C API to libtorch")