[gentoo-commits] repo/gentoo:master commit in: sci-libs/pytorch/

[Alfredo Tupone]

commit:     78893e494fae29e990504d527800096efb48299e
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Tue Jul  5 06:08:24 2022 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Tue Jul  5 06:08:24 2022 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=78893e49

sci-libs/pytorch: add USE flags controlling caffe2

Package-Manager: Portage-3.0.30, Repoman-3.0.3
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-libs/pytorch/metadata.xml          | 11 +++++++++++
 sci-libs/pytorch/pytorch-1.11.0.ebuild |  2 ++
 2 files changed, 13 insertions(+)

diff --git a/sci-libs/pytorch/metadata.xml b/sci-libs/pytorch/metadata.xml
index d12749aa5c21..bc2785e5f6db 100644
--- a/sci-libs/pytorch/metadata.xml
+++ b/sci-libs/pytorch/metadata.xml
@@ -5,6 +5,17 @@
                <email>tup...@gentoo.org</email>
                <name>Tupone Alfredo</name>
        </maintainer>
+       <use>
+               <flag name="cuda">Add support for CUDA processing</flag>
+               <flag name="ffmpeg">Add support for video processing 
operators</flag>
+               <flag name="nnpack">Use NNPACK</flag>
+               <flag name="numpy">Add support for math operations through 
numpy</flag>
+               <flag name="opencl">Use OpenCL</flag>
+               <flag name="opencv">Add support for image processing 
operators</flag>
+               <flag name="openmp">Use OpenMP for parallel code</flag>
+               <flag name="qnnpack">Use QNNPACK</flag>
+               <flag name="xnnpack">Use XNNPACK</flag>
+       </use>
        <upstream>
                <remote-id type="github">pytorch/pytorch</remote-id>
        </upstream>

diff --git a/sci-libs/pytorch/pytorch-1.11.0.ebuild 
b/sci-libs/pytorch/pytorch-1.11.0.ebuild
index 0a1cae78f4bb..401bdea8264a 100644
--- a/sci-libs/pytorch/pytorch-1.11.0.ebuild
+++ b/sci-libs/pytorch/pytorch-1.11.0.ebuild
@@ -16,11 +16,13 @@ LICENSE="BSD"
 SLOT="0"
 KEYWORDS="~amd64"
 RESTRICT="test"
+IUSE="cuda ffmpeg nnpack +numpy opencl opencv openmp qnnpack xnnpack"
 
 REQUIRED_USE=${PYTHON_REQUIRED_USE}
 RDEPEND="
        ${PYTHON_DEPS}
        ~sci-libs/caffe2-${PV}[${PYTHON_USEDEP}]
+       
sci-libs/caffe2[cuda?,ffmpeg?,nnpack?,numpy?,opencl?,opencv?,openmp?,qnnpack?,xnnpack?]
        dev-python/typing-extensions[${PYTHON_USEDEP}]
 "
 DEPEND="${RDEPEND}