commit: 70a1cabccd1d81a307686ab693d0443a439b64d1
Author: Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Tue Feb 28 07:10:59 2023 +0000
Commit: Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Tue Feb 28 07:12:05 2023 +0000
URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=70a1cabc
sci-libs/caffe2: add mpi use flag
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>
.../caffe2/{caffe2-1.13.1-r2.ebuild => caffe2-1.13.1-r3.ebuild} | 8 ++++++--
sci-libs/caffe2/metadata.xml | 1 +
2 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/sci-libs/caffe2/caffe2-1.13.1-r2.ebuild
b/sci-libs/caffe2/caffe2-1.13.1-r3.ebuild
similarity index 95%
rename from sci-libs/caffe2/caffe2-1.13.1-r2.ebuild
rename to sci-libs/caffe2/caffe2-1.13.1-r3.ebuild
index d985af6d9995..04521a5c26ef 100644
--- a/sci-libs/caffe2/caffe2-1.13.1-r2.ebuild
+++ b/sci-libs/caffe2/caffe2-1.13.1-r3.ebuild
@@ -17,11 +17,12 @@
SRC_URI="https://github.com/pytorch/${MYPN}/archive/refs/tags/v${PV}.tar.gz
LICENSE="BSD"
SLOT="0"
KEYWORDS="~amd64"
-IUSE="cuda ffmpeg nnpack +numpy opencl opencv openmp qnnpack xnnpack"
+IUSE="cuda distributed ffmpeg mpi nnpack +numpy opencl opencv openmp qnnpack
xnnpack"
RESTRICT="test"
REQUIRED_USE="
${PYTHON_REQUIRED_USE}
ffmpeg? ( opencv )
+ mpi? ( distributed )
" # ?? ( cuda rocm )
# CUDA 12 not supported yet: https://github.com/pytorch/pytorch/issues/91122
@@ -43,6 +44,7 @@ RDEPEND="
<dev-util/nvidia-cuda-toolkit-12:=[profiler]
)
ffmpeg? ( media-video/ffmpeg:= )
+ mpi? ( sys-cluster/openmpi )
nnpack? ( sci-libs/NNPACK )
numpy? ( $(python_gen_cond_dep '
dev-python/numpy[${PYTHON_USEDEP}]
@@ -108,7 +110,8 @@ src_configure() {
-DUSE_CUDNN=$(usex cuda)
-DUSE_FAST_NVCC=$(usex cuda)
-DTORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5 7.0}"
- -DUSE_DISTRIBUTED=OFF
+ -DUSE_DISTRIBUTED=$(usex distributed)
+ -DUSE_MPI=$(usex mpi)
-DUSE_FAKELOWP=OFF
-DUSE_FBGEMM=OFF # TODO
-DUSE_FFMPEG=$(usex ffmpeg)
@@ -132,6 +135,7 @@ src_configure() {
-DUSE_ROCM=OFF # TODO
-DUSE_SYSTEM_CPUINFO=ON
-DUSE_SYSTEM_PYBIND11=ON
+ -DUSE_UCC=OFF
-DUSE_VALGRIND=OFF
-DPYBIND11_PYTHON_VERSION="${EPYTHON#python}"
-DPYTHON_EXECUTABLE="${PYTHON}"
diff --git a/sci-libs/caffe2/metadata.xml b/sci-libs/caffe2/metadata.xml
index 25a171ff1680..f009baee81bb 100644
--- a/sci-libs/caffe2/metadata.xml
+++ b/sci-libs/caffe2/metadata.xml
@@ -11,6 +11,7 @@
</maintainer>
<use>
<flag name="cuda">Add support for CUDA processing</flag>
+ <flag name="distributed">Support distributed applications</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>
