From 09e78c34af071f1c0b09d673ed243829739a9239 Mon Sep 17 00:00:00 2001 From: David Elsing Date: Wed, 22 Jan 2025 18:32:23 +0000 Subject: [PATCH] gnu: python-pytorch-for-r-torch: Fix build. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The build fails because the types of function pointers with return type '__m256' and 'const __m256' (or '__m512' and 'const __m512') are not treated as compatible by the compiler. * gnu/packages/machine-learning.scm (python-pytorch-for-r-torch) [arguments]: New field. [inputs]: Add foxi. Signed-off-by: Ludovic Courtès --- gnu/packages/machine-learning.scm | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index 39defe62714c7c676dcdb640b560b998e66e2062..66d076690c01d5e28adeabf8489a2f9d5ed2bc0b 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -5128,11 +5128,23 @@ Note: currently this package does not provide GPU support.") (name "python-pytorch") (version %python-pytorch-for-r-torch-version) (source %python-pytorch-for-r-torch-src) + (arguments + (substitute-keyword-arguments (package-arguments python-pytorch) + ((#:phases phases) + #~(modify-phases #$phases + ;; See https://github.com/pytorch/pytorch/issues/61244 + (add-after 'unpack 'fix-aten-vec + (lambda _ + (substitute* + '("aten/src/ATen/cpu/vec/vec512/vec512_bfloat16.h" + "aten/src/ATen/cpu/vec/vec256/vec256_bfloat16.h") + (("map\\(const __") "map(__")))))))) (native-inputs (modify-inputs (package-native-inputs python-pytorch) (replace "ideep-pytorch" ideep-pytorch-for-r-torch))) (inputs (modify-inputs (package-inputs python-pytorch) + (prepend foxi) (prepend qnnpack) (replace "qnnpack-pytorch" qnnpack-pytorch-for-r-torch) (replace "oneapi-dnnl" oneapi-dnnl-for-r-torch)