me
/
guix
Archived
1
0
Fork 0

gnu: Add xnnpack-for-torch2.

* gnu/packages/machine-learning.scm (xnnpack-for-torch2): New variable.
* gnu/packages/patches/xnnpack-for-torch2-system-libraries.patch: New file.
* gnu/local.mk (dist_patch_DATA): Add it.

Change-Id: I1c35f5fbd064fcc6732b924d7cafaa1a7d03ce04
master
Ricardo Wurmus 2023-12-27 13:47:10 +01:00
parent 71b8ed4a28
commit 5654f61541
No known key found for this signature in database
GPG Key ID: 197A5888235FACAC
3 changed files with 2689 additions and 0 deletions

View File

@ -2161,6 +2161,7 @@ dist_patch_DATA = \
%D%/packages/patches/xgboost-use-system-dmlc-core.patch \
%D%/packages/patches/xmonad-dynamic-linking.patch \
%D%/packages/patches/xnnpack-system-libraries.patch \
%D%/packages/patches/xnnpack-for-torch2-system-libraries.patch \
%D%/packages/patches/xplanet-1.3.1-cxx11-eof.patch \
%D%/packages/patches/xplanet-1.3.1-libdisplay_DisplayOutput.cpp.patch \
%D%/packages/patches/xplanet-1.3.1-libimage_gif.c.patch \

View File

@ -3878,6 +3878,34 @@ high-level machine learning frameworks, such as TensorFlow Lite,
TensorFlow.js, PyTorch, and MediaPipe.")
(license license:bsd-3))))
(define-public xnnpack-for-torch2
;; There's currently no tag on this repo.
(let ((version "0.0")
(commit "51a987591a6fc9f0fc0707077f53d763ac132cbf")
(revision "3"))
(package
(inherit xnnpack)
(name "xnnpack")
(version (git-version version revision commit))
(home-page "https://github.com/google/XNNPACK") ;fork of QNNPACK
(source (origin
(method git-fetch)
(uri (git-reference (url home-page) (commit commit)))
(file-name (git-file-name name version))
(sha256
(base32
"1rzby82xq8d0rl1d148yz88jh9cpsw5c8b2yw7yg39mi7qmr55rm"))
(patches (search-patches "xnnpack-for-torch2-system-libraries.patch"))))
(arguments
(list
#:tests? #false
#:configure-flags '(list "-DXNNPACK_USE_SYSTEM_LIBS=YES"
"-DBUILD_SHARED_LIBS=ON"
"-DCMAKE_POSITION_INDEPENDENT_CODE=ON"
"-DXNNPACK_LIBRARY_TYPE=shared"
"-DXNNPACK_BUILD_TESTS=FALSE" ;FIXME: see below
"-DXNNPACK_BUILD_BENCHMARKS=FALSE"))))))
;; Please also update python-torchvision when updating this package.
(define-public python-pytorch
(package

File diff suppressed because it is too large Load Diff