From patchwork Wed Mar 20 22:38:52 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: David Elsing X-Patchwork-Id: 62096 Return-Path: X-Original-To: patchwork@mira.cbaines.net Delivered-To: patchwork@mira.cbaines.net Received: by mira.cbaines.net (Postfix, from userid 113) id 6A25927BBE2; Wed, 20 Mar 2024 22:41:47 +0000 (GMT) X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on mira.cbaines.net X-Spam-Level: X-Spam-Status: No, score=-2.7 required=5.0 tests=BAYES_00,DKIM_INVALID, DKIM_SIGNED,MAILING_LIST_MULTI,SPF_HELO_PASS,URIBL_BLOCKED autolearn=ham autolearn_force=no version=3.4.6 Received: from lists.gnu.org (lists.gnu.org [209.51.188.17]) by mira.cbaines.net (Postfix) with ESMTPS id F3AAC27BBE2 for ; Wed, 20 Mar 2024 22:41:46 +0000 (GMT) Received: from localhost ([::1] helo=lists1p.gnu.org) by lists.gnu.org with esmtp (Exim 4.90_1) (envelope-from ) id 1rn4bq-0005yf-LP; Wed, 20 Mar 2024 18:40:34 -0400 Received: from eggs.gnu.org ([2001:470:142:3::10]) by lists.gnu.org with esmtps (TLS1.2:ECDHE_RSA_AES_256_GCM_SHA384:256) (Exim 4.90_1) (envelope-from ) id 1rn4bm-0005xK-SQ for guix-patches@gnu.org; Wed, 20 Mar 2024 18:40:30 -0400 Received: from debbugs.gnu.org ([2001:470:142:5::43]) by eggs.gnu.org with esmtps (TLS1.2:ECDHE_RSA_AES_128_GCM_SHA256:128) (Exim 4.90_1) (envelope-from ) id 1rn4bm-0007cZ-Il for guix-patches@gnu.org; Wed, 20 Mar 2024 18:40:30 -0400 Received: from Debian-debbugs by debbugs.gnu.org with local (Exim 4.84_2) (envelope-from ) id 1rn4cO-0006Dn-Tp for guix-patches@gnu.org; Wed, 20 Mar 2024 18:41:08 -0400 X-Loop: help-debbugs@gnu.org Subject: [bug#69591] [PATCH v3 27/32] gnu: Add python-pytorch-avx. Resent-From: David Elsing Original-Sender: "Debbugs-submit" Resent-CC: guix-patches@gnu.org Resent-Date: Wed, 20 Mar 2024 22:41:08 +0000 Resent-Message-ID: Resent-Sender: help-debbugs@gnu.org X-GNU-PR-Message: followup 69591 X-GNU-PR-Package: guix-patches X-GNU-PR-Keywords: patch To: 69591@debbugs.gnu.org Cc: David Elsing Received: via spool by 69591-submit@debbugs.gnu.org id=B69591.171097443223598 (code B ref 69591); Wed, 20 Mar 2024 22:41:08 +0000 Received: (at 69591) by debbugs.gnu.org; 20 Mar 2024 22:40:32 +0000 Received: from localhost ([127.0.0.1]:36815 helo=debbugs.gnu.org) by debbugs.gnu.org with esmtp (Exim 4.84_2) (envelope-from ) id 1rn4bk-00068P-Db for submit@debbugs.gnu.org; Wed, 20 Mar 2024 18:40:31 -0400 Received: from mout02.posteo.de ([185.67.36.66]:57857) by debbugs.gnu.org with esmtp (Exim 4.84_2) (envelope-from ) id 1rn4bN-00062x-LK for 69591@debbugs.gnu.org; Wed, 20 Mar 2024 18:40:07 -0400 Received: from submission (posteo.de [185.67.36.169]) by mout02.posteo.de (Postfix) with ESMTPS id A1D76240103 for <69591@debbugs.gnu.org>; Wed, 20 Mar 2024 23:39:25 +0100 (CET) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=posteo.net; s=2017; t=1710974365; bh=WEsaR5AaZr02LZx+IjGRIsTTgwzCQ7Irb5p0ROEOU7A=; h=From:To:Cc:Subject:Date:Message-ID:MIME-Version: Content-Transfer-Encoding:From; b=dfbADBT+VmAb5mZj4Gbz0Xq+qazAJm0NCSoEAYqZxhRcHeo+cti2BDOaFRfXlwOBd UZNEmGJl8jhgFm05zErH+47n6bT6eKKgttIp/uHZvmYvZkV2kzhzRqhUX49ZS8U0J6 hrHzx/mKlJYLsERk7xIVfYOkyGE8DiBnbTDRtYvmN0CIdls7VCQ4t0IBAjTiGMx8aW kBZy6OwRUmNow7twMe5yBGTti9UkJI6YhztSHX21zfqr7967Y9no6WMj8nOxMXMIht 5Lq1cnu1W8u7pzQMdom7af32uzNsEN1MsG9CW2f2nZGhKv/gK9FaaBv+1QByp/R9iA 5FkI8piV0MZAA== Received: from customer (localhost [127.0.0.1]) by submission (posteo.de) with ESMTPSA id 4V0Nn92S51z6twY; Wed, 20 Mar 2024 23:39:25 +0100 (CET) From: David Elsing Date: Wed, 20 Mar 2024 22:38:52 +0000 Message-ID: <20240320223906.13214-27-david.elsing@posteo.net> In-Reply-To: <20240320223737.12883-1-david.elsing@posteo.net> References: <20240320223737.12883-1-david.elsing@posteo.net> MIME-Version: 1.0 X-BeenThere: debbugs-submit@debbugs.gnu.org X-Mailman-Version: 2.1.18 Precedence: list X-BeenThere: guix-patches@gnu.org List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Errors-To: guix-patches-bounces+patchwork=mira.cbaines.net@gnu.org Sender: guix-patches-bounces+patchwork=mira.cbaines.net@gnu.org X-getmail-retrieved-from-mailbox: Patches * gnu/packages/machine-learning.scm (python-pytorch-avx): New variable. --- gnu/packages/machine-learning.scm | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index a062a2398b..ea023ad749 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -4712,6 +4712,21 @@ (define-public python-pytorch Note: currently this package does not provide GPU support.") (license license:bsd-3))) +;; This package variant includes the dependencies requiring at least AVX2 or +;; AVX-512. +(define-public python-pytorch-avx + (package/inherit python-pytorch + (name "python-pytorch-avx") + (inputs + (modify-inputs (package-inputs python-pytorch) + (append fbgemm nnpack))) + (arguments + (substitute-keyword-arguments (package-arguments python-pytorch) + ((#:phases phases) + #~(modify-phases #$phases + (delete 'disable-avx-dependencies))))) + (supported-systems '("x86_64-linux")))) + (define-public python-pytorch-for-r-torch (package (inherit python-pytorch)