gnu: Add onnx-for-torch2.

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

Change-Id: I8a46b9a8096a33bcf949bc4f5d497e22902ab301
This commit is contained in:
Ricardo Wurmus 2023-12-27 16:13:30 +01:00
parent 7720430678
commit 43c92f323e
No known key found for this signature in database
GPG key ID: 197A5888235FACAC
3 changed files with 79 additions and 0 deletions

View file

@ -1214,6 +1214,29 @@ an extensible computation graph model, as well as definitions of built-in
operators and standard data types.")
(license license:expat)))
(define-public onnx-for-torch2
(package
(inherit onnx)
(name "onnx")
(version "1.13.1")
(source (origin
(method git-fetch)
(uri (git-reference
(url "https://github.com/onnx/onnx")
(commit (string-append "v" version))))
(sha256
(base32
"16967dbq2j40diqd0s37r19llsab8q8vbxkg1ppgy0p9fpdhfhyp"))
(file-name (git-file-name name version))
(patches (search-patches "onnx-1.13.1-use-system-googletest.patch"
"onnx-shared-libraries.patch"))
(modules '((guix build utils)))
(snippet
'(begin
(delete-file-recursively "third_party")
(substitute* "onnx/backend/test/runner/__init__.py"
(("urlretrieve\\(.*") "raise unittest.SkipTest('Skipping download')\n"))))))))
(define-public python-onnx
;; This used to be called "python-onnx" because it provided nothing but
;; Python bindings. The package now provides shared libraries and C++