diff options
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.1.2-r4.ebuild (renamed from sci-libs/caffe2/caffe2-2.1.2-r3.ebuild) | 5 | ||||
| -rw-r--r-- | sci-libs/caffe2/metadata.xml | 3 |
2 files changed, 6 insertions, 2 deletions
diff --git a/sci-libs/caffe2/caffe2-2.1.2-r3.ebuild b/sci-libs/caffe2/caffe2-2.1.2-r4.ebuild index 9b7554309dbc..e4d9ad2932f3 100644 --- a/sci-libs/caffe2/caffe2-2.1.2-r3.ebuild +++ b/sci-libs/caffe2/caffe2-2.1.2-r4.ebuild @@ -19,7 +19,7 @@ onednn? ( https://github.com/intel/ideep/archive/${IDEEP_VERSION}.tar.gz -> idee LICENSE="BSD" SLOT="0" KEYWORDS="~amd64" -IUSE="cuda distributed fbgemm ffmpeg gloo mkl mpi nnpack +numpy onednn opencl opencv openmp qnnpack tensorpipe xnnpack" +IUSE="cuda distributed fbgemm ffmpeg gloo mkl mpi nnpack +numpy onednn openblas opencl opencv openmp qnnpack tensorpipe xnnpack" RESTRICT="test" REQUIRED_USE=" ${PYTHON_REQUIRED_USE} @@ -64,6 +64,7 @@ RDEPEND=" tensorpipe? ( sci-libs/tensorpipe[cuda?] ) xnnpack? ( >=sci-libs/XNNPACK-2022.12.22 ) mkl? ( sci-libs/mkl ) + openblas? ( sci-libs/openblas ) " DEPEND=" ${RDEPEND} @@ -189,6 +190,8 @@ src_configure() { if use mkl; then mycmakeargs+=(-DBLAS=MKL) + elif use openblas; then + mycmakeargs+=(-DBLAS=OpenBLAS) else mycmakeargs+=(-DBLAS=Generic -DBLAS_LIBRARIES=) fi diff --git a/sci-libs/caffe2/metadata.xml b/sci-libs/caffe2/metadata.xml index 64d212edd735..3fe84b0977fc 100644 --- a/sci-libs/caffe2/metadata.xml +++ b/sci-libs/caffe2/metadata.xml @@ -10,15 +10,16 @@ <flag name="fbgemm">Use FBGEMM</flag> <flag name="ffmpeg">Add support for video processing operators</flag> <flag name="gloo">Use sci-libs/gloo</flag> + <flag name="mkl">Use <pkg>sci-libs/mkl</pkg> for blas, lapack and sparse blas routines</flag> <flag name="nnpack">Use NNPACK</flag> <flag name="numpy">Add support for math operations through numpy</flag> <flag name="onednn">Use oneDNN</flag> + <flag name="openblas">Use <pkg>sci-libs/openblas</pkg> for blas routines</flag> <flag name="opencv">Add support for image processing operators</flag> <flag name="openmp">Use OpenMP for parallel code</flag> <flag name="qnnpack">Use QNNPACK</flag> <flag name="tensorpipe">Use tensorpipe</flag> <flag name="xnnpack">Use XNNPACK</flag> - <flag name="mkl">Use <pkg>sci-libs/mkl</pkg> for blas, lapack and sparse blas routines</flag> </use> <upstream> <remote-id type="github">pytorch/pytorch</remote-id> |