diff options
| author | David Elsing <david.elsing@posteo.net> | 2025-01-22 18:32:22 +0000 |
|---|---|---|
| committer | Ludovic Courtès <ludo@gnu.org> | 2025-01-27 14:42:40 +0100 |
| commit | a0a9044d356295be45841add1c239d377ebb3ddf (patch) | |
| tree | 2d52bb39b6bb9a35b516767e9e973a3676c4bd84 /gnu/packages/machine-learning.scm | |
| parent | 9a6221e76ccffe72f908ca4f03e1e3f83d2359c7 (diff) | |
gnu: python-pytorch: Update to 2.5.1.
This also updates the qnnpack-pytorch package.
* gnu/packages/machine-learning.scm (%python-pytorch-version): Set to 2.5.1.
(%python-pytorch-src): Adjust hash accordingly.
(python-pytorch)[arguments]<#:phases>: Add 'remove-fr-trace-script
phase. Adjust 'use-system-libraries phase.
[native-inputs]: Add nlohmann-json. Replace cmake with cmake-minimal.
[inputs]: Remove foxi.
* gnu/packages/patches/python-pytorch-fix-codegen.patch: Adjust patch.
* gnu/packages/patches/python-pytorch-system-libraries.patch: Adjust patch.
Signed-off-by: Ludovic Courtès <ludo@gnu.org>
Diffstat (limited to 'gnu/packages/machine-learning.scm')
| -rw-r--r-- | gnu/packages/machine-learning.scm | 21 |
1 files changed, 16 insertions, 5 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index 3917f3a79e..39defe6271 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -4670,7 +4670,7 @@ PyTorch.") (base32 "0hdpkhcjry22fjx2zg2r48v7f4ljrclzj0li2pgk76kvyblfbyvm")))))) -(define %python-pytorch-version "2.4.0") +(define %python-pytorch-version "2.5.1") (define %python-pytorch-src (origin @@ -4681,7 +4681,7 @@ PyTorch.") (file-name (git-file-name "python-pytorch" %python-pytorch-version)) (sha256 (base32 - "18hdhzr12brj0b7ppyiscax0dbra30207qx0cckw78midfkcn7cn")) + "052cvagpmm9y7jspjpcyysx8yc5fhxnjl8rcz6nndis06v8dcj8s")) (patches (search-patches "python-pytorch-system-libraries.patch" "python-pytorch-runpath.patch" "python-pytorch-without-kineto.patch" @@ -4827,6 +4827,14 @@ PyTorch.") (string-append #$output "/lib/python" #$(version-major+minor (package-version python)) "/site-packages"))))) + ;; This entry point is broken, because it refers to a module that is + ;; (intentionally) not installed + ;; (https://github.com/pytorch/pytorch/pull/134729), which causes + ;; the 'sanity-check phase to fail. + (add-after 'unpack 'remove-fr-trace-script + (lambda _ + (substitute* "setup.py" + (("entry_points\\[\"console_scripts\"\\]\\.append\\(") "(")))) (add-before 'build 'use-system-libraries (lambda _ (substitute* '("caffe2/serialize/crc.cc" @@ -4864,7 +4872,10 @@ PyTorch.") (or (%current-target-system) (%current-system)) (package-transitive-supported-systems qnnpack))) - (setenv "USE_QNNPACK" "0")))) + (setenv "USE_QNNPACK" "0")) + (substitute* '("requirements.txt" "setup.py") + (("sympy==1\\.13\\.1") + "sympy>=1.13.1")))) ;; PyTorch is still built with AVX2 and AVX-512 support selected at ;; runtime, but these dependencies require it (nnpack only for ;; x86_64). @@ -4965,10 +4976,11 @@ PyTorch.") ;; Even only the core tests take a very long time to run. #:tests? #f)) (native-inputs - (list cmake + (list cmake-minimal doxygen ideep-pytorch ninja + nlohmann-json pocketfft-cpp python-expecttest python-pytest-flakefinder @@ -4988,7 +5000,6 @@ PyTorch.") eigen flatbuffers-next fmt - foxi fp16 fxdiv gemmlowp |