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

[Alfredo Tupone]

commit:     2299e56b110834ccc6e8d9a675d1093a4d8559a0
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Sat Apr  8 14:28:31 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Sat Apr  8 14:28:31 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=2299e56b

sci-libs/caffe2: enable use of FBGEMM

Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-libs/caffe2/{caffe2-2.0.0.ebuild => caffe2-2.0.0-r1.ebuild} | 5 +++--
 sci-libs/caffe2/metadata.xml                                    | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/sci-libs/caffe2/caffe2-2.0.0.ebuild 
b/sci-libs/caffe2/caffe2-2.0.0-r1.ebuild
similarity index 96%
rename from sci-libs/caffe2/caffe2-2.0.0.ebuild
rename to sci-libs/caffe2/caffe2-2.0.0-r1.ebuild
index 9b010cdda9a9..ae14175b2471 100644
--- a/sci-libs/caffe2/caffe2-2.0.0.ebuild
+++ b/sci-libs/caffe2/caffe2-2.0.0-r1.ebuild
@@ -17,7 +17,7 @@ 
SRC_URI="https://github.com/pytorch/${MYPN}/archive/refs/tags/v${PV}.tar.gz
 LICENSE="BSD"
 SLOT="0"
 KEYWORDS="~amd64"
-IUSE="cuda distributed ffmpeg mpi nnpack +numpy opencl opencv openmp qnnpack 
tensorpipe xnnpack"
+IUSE="cuda distributed fbgemm ffmpeg mpi nnpack +numpy opencl opencv openmp 
qnnpack tensorpipe xnnpack"
 RESTRICT="test"
 REQUIRED_USE="
        ${PYTHON_REQUIRED_USE}
@@ -44,6 +44,7 @@ RDEPEND="
                dev-libs/cudnn-frontend:0/8
                <dev-util/nvidia-cuda-toolkit-12:=[profiler]
        )
+       fbgemm? ( dev-libs/FBGEMM )
        ffmpeg? ( media-video/ffmpeg:= )
        mpi? ( sys-cluster/openmpi )
        nnpack? ( sci-libs/NNPACK )
@@ -115,7 +116,7 @@ src_configure() {
                -DUSE_DISTRIBUTED=$(usex distributed)
                -DUSE_MPI=$(usex mpi)
                -DUSE_FAKELOWP=OFF
-               -DUSE_FBGEMM=OFF # TODO
+               -DUSE_FBGEMM=$(usex fbgemm)
                -DUSE_FFMPEG=$(usex ffmpeg)
                -DUSE_GFLAGS=ON
                -DUSE_GLOG=ON

diff --git a/sci-libs/caffe2/metadata.xml b/sci-libs/caffe2/metadata.xml
index 531b9cbf5879..f58d3b5e97ab 100644
--- a/sci-libs/caffe2/metadata.xml
+++ b/sci-libs/caffe2/metadata.xml
@@ -12,6 +12,7 @@
        <use>
                <flag name="cuda">Add support for CUDA processing</flag>
                <flag name="distributed">Support distributed applications</flag>
+               <flag name="fbgemm">Use FBGEMM</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>