1 |
commit: 78893e494fae29e990504d527800096efb48299e |
2 |
Author: Alfredo Tupone <tupone <AT> gentoo <DOT> org> |
3 |
AuthorDate: Tue Jul 5 06:08:24 2022 +0000 |
4 |
Commit: Alfredo Tupone <tupone <AT> gentoo <DOT> org> |
5 |
CommitDate: Tue Jul 5 06:08:24 2022 +0000 |
6 |
URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=78893e49 |
7 |
|
8 |
sci-libs/pytorch: add USE flags controlling caffe2 |
9 |
|
10 |
Package-Manager: Portage-3.0.30, Repoman-3.0.3 |
11 |
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org> |
12 |
|
13 |
sci-libs/pytorch/metadata.xml | 11 +++++++++++ |
14 |
sci-libs/pytorch/pytorch-1.11.0.ebuild | 2 ++ |
15 |
2 files changed, 13 insertions(+) |
16 |
|
17 |
diff --git a/sci-libs/pytorch/metadata.xml b/sci-libs/pytorch/metadata.xml |
18 |
index d12749aa5c21..bc2785e5f6db 100644 |
19 |
--- a/sci-libs/pytorch/metadata.xml |
20 |
+++ b/sci-libs/pytorch/metadata.xml |
21 |
@@ -5,6 +5,17 @@ |
22 |
<email>tupone@g.o</email> |
23 |
<name>Tupone Alfredo</name> |
24 |
</maintainer> |
25 |
+ <use> |
26 |
+ <flag name="cuda">Add support for CUDA processing</flag> |
27 |
+ <flag name="ffmpeg">Add support for video processing operators</flag> |
28 |
+ <flag name="nnpack">Use NNPACK</flag> |
29 |
+ <flag name="numpy">Add support for math operations through numpy</flag> |
30 |
+ <flag name="opencl">Use OpenCL</flag> |
31 |
+ <flag name="opencv">Add support for image processing operators</flag> |
32 |
+ <flag name="openmp">Use OpenMP for parallel code</flag> |
33 |
+ <flag name="qnnpack">Use QNNPACK</flag> |
34 |
+ <flag name="xnnpack">Use XNNPACK</flag> |
35 |
+ </use> |
36 |
<upstream> |
37 |
<remote-id type="github">pytorch/pytorch</remote-id> |
38 |
</upstream> |
39 |
|
40 |
diff --git a/sci-libs/pytorch/pytorch-1.11.0.ebuild b/sci-libs/pytorch/pytorch-1.11.0.ebuild |
41 |
index 0a1cae78f4bb..401bdea8264a 100644 |
42 |
--- a/sci-libs/pytorch/pytorch-1.11.0.ebuild |
43 |
+++ b/sci-libs/pytorch/pytorch-1.11.0.ebuild |
44 |
@@ -16,11 +16,13 @@ LICENSE="BSD" |
45 |
SLOT="0" |
46 |
KEYWORDS="~amd64" |
47 |
RESTRICT="test" |
48 |
+IUSE="cuda ffmpeg nnpack +numpy opencl opencv openmp qnnpack xnnpack" |
49 |
|
50 |
REQUIRED_USE=${PYTHON_REQUIRED_USE} |
51 |
RDEPEND=" |
52 |
${PYTHON_DEPS} |
53 |
~sci-libs/caffe2-${PV}[${PYTHON_USEDEP}] |
54 |
+ sci-libs/caffe2[cuda?,ffmpeg?,nnpack?,numpy?,opencl?,opencv?,openmp?,qnnpack?,xnnpack?] |
55 |
dev-python/typing-extensions[${PYTHON_USEDEP}] |
56 |
" |
57 |
DEPEND="${RDEPEND} |