mirror of
https://git.in.rschanz.org/ryan77627/guix.git
synced 2025-01-24 11:39:46 -05:00
gnu: Add tensorpipe.
* gnu/packages/machine-learning.scm (tensorpipe): New variable.
This commit is contained in:
parent
a039f0fc59
commit
c45b370385
1 changed files with 36 additions and 0 deletions
|
@ -91,6 +91,7 @@ (define-module (gnu packages machine-learning)
|
|||
#:use-module (gnu packages image-processing)
|
||||
#:use-module (gnu packages imagemagick)
|
||||
#:use-module (gnu packages jupyter)
|
||||
#:use-module (gnu packages libevent)
|
||||
#:use-module (gnu packages libffi)
|
||||
#:use-module (gnu packages linux)
|
||||
#:use-module (gnu packages llvm)
|
||||
|
@ -4114,6 +4115,41 @@ (define-public xnnpack-for-torch2
|
|||
"-DXNNPACK_BUILD_TESTS=FALSE" ;FIXME: see below
|
||||
"-DXNNPACK_BUILD_BENCHMARKS=FALSE"))))))
|
||||
|
||||
(define-public tensorpipe
|
||||
(let ((commit "bb1473a4b38b18268e8693044afdb8635bc8351b")
|
||||
(revision "0"))
|
||||
(package
|
||||
(name "tensorpipe")
|
||||
(version (git-version "0" revision commit))
|
||||
(source (origin
|
||||
(method git-fetch)
|
||||
(uri (git-reference
|
||||
(url "https://github.com/pytorch/tensorpipe")
|
||||
(commit commit)))
|
||||
(file-name (git-file-name name version))
|
||||
(sha256
|
||||
(base32
|
||||
"0sbpkd69rzybw2j89sjkf4s0j8vkk96d51bsps28894989a75j6v"))
|
||||
(modules '((guix build utils)))
|
||||
(snippet
|
||||
'(delete-file-recursively "third_party"))))
|
||||
(build-system cmake-build-system)
|
||||
(arguments
|
||||
(list
|
||||
#:configure-flags
|
||||
''("-DBUILD_SHARED_LIBS=ON")
|
||||
;; There are no tests
|
||||
#:tests? #f))
|
||||
(inputs (list libuv))
|
||||
(native-inputs (list googletest pkg-config pybind11 libnop))
|
||||
(home-page "https://github.com/pytorch/tensorpipe")
|
||||
(synopsis "Tensor-aware point-to-point communication primitive for
|
||||
machine learning")
|
||||
(description "TensorPipe provides a tensor-aware channel to transfer
|
||||
rich objects from one process to another while using the fastest transport for
|
||||
the tensors contained therein.")
|
||||
(license license:bsd-3))))
|
||||
|
||||
;; Please also update python-torchvision when updating this package.
|
||||
(define-public python-pytorch
|
||||
(package
|
||||
|
|
Loading…
Reference in a new issue