diff options
| author |   Alfredo Tupone <tupone@gentoo.org> | 2023-02-28 08:10:59 +0100 |
|---|---|---|
| committer | Alfredo Tupone <tupone@gentoo.org> | 2023-02-28 08:12:05 +0100 |
| commit | 70a1cabccd1d81a307686ab693d0443a439b64d1 (patch) | |
| tree | 4a6bc3e092908001a2a7b641733d6a63d30e1175 /sci-libs/caffe2 | |
| parent | sci-libs/caffe2: CUDA 12 not supported yet (diff) | |
| download | gentoo-70a1cabccd1d81a307686ab693d0443a439b64d1.tar.gz gentoo-70a1cabccd1d81a307686ab693d0443a439b64d1.tar.bz2 gentoo-70a1cabccd1d81a307686ab693d0443a439b64d1.zip | |
sci-libs/caffe2: add mpi use flag
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs/caffe2')
| -rw-r--r-- | sci-libs/caffe2/caffe2-1.13.1-r3.ebuild (renamed from sci-libs/caffe2/caffe2-1.13.1-r2.ebuild) | 8 | ||||
| -rw-r--r-- | 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 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> |