diff options
| author |   Alfredo Tupone <tupone@gentoo.org> | 2023-04-08 16:28:31 +0200 |
|---|---|---|
| committer | Alfredo Tupone <tupone@gentoo.org> | 2023-04-08 16:28:31 +0200 |
| commit | 2299e56b110834ccc6e8d9a675d1093a4d8559a0 (patch) | |
| tree | 9c1acf1ada884f2ccafb076c0c5a44da48a17e1c /sci-libs | |
| parent | dev-libs/FBGEMM: add 2023.02.10, remove old (diff) | |
| download | gentoo-2299e56b110834ccc6e8d9a675d1093a4d8559a0.tar.gz gentoo-2299e56b110834ccc6e8d9a675d1093a4d8559a0.tar.bz2 gentoo-2299e56b110834ccc6e8d9a675d1093a4d8559a0.zip | |
sci-libs/caffe2: enable use of FBGEMM
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs')
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.0.0-r1.ebuild (renamed from sci-libs/caffe2/caffe2-2.0.0.ebuild) | 5 | ||||
| -rw-r--r-- | 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 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> |