diff options
| author | David Elsing <david.elsing@posteo.net> | 2024-03-23 22:04:57 +0000 |
|---|---|---|
| committer | Ludovic Courtès <ludo@gnu.org> | 2024-06-18 11:42:32 +0200 |
| commit | c45b370385e35a9d6444b16017cf08c97e1463ae (patch) | |
| tree | 9ee4f8d3a40d060bbfbedbc36d0d255319ed6f53 /gnu/packages/machine-learning.scm | |
| parent | a039f0fc59975af534d7550a0d5655b4418795da (diff) | |
| download | guix-patches-c45b370385e35a9d6444b16017cf08c97e1463ae.tar guix-patches-c45b370385e35a9d6444b16017cf08c97e1463ae.tar.gz | |
gnu: Add tensorpipe.
* gnu/packages/machine-learning.scm (tensorpipe): New variable.
Diffstat (limited to 'gnu/packages/machine-learning.scm')
| -rw-r--r-- | gnu/packages/machine-learning.scm | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index 8e9ad5ca0e..ddeb9a166a 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -91,6 +91,7 @@ #: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 @@ TensorFlow.js, PyTorch, and MediaPipe.") "-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 |