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>