diff options
| author |   Alfredo Tupone <tupone@gentoo.org> | 2024-05-09 20:44:58 +0200 |
|---|---|---|
| committer | Alfredo Tupone <tupone@gentoo.org> | 2024-05-09 20:45:45 +0200 |
| commit | 27e4c06d52a66978ec41a325468bdd3b756236ba (patch) | |
| tree | fd28855806bd3931ec3b162363f604c1c726baf7 /sci-libs | |
| parent | dev-util/gdbus-codegen: Stabilize 2.78.6 ppc64, #931633 (diff) | |
| download | gentoo-27e4c06d52a66978ec41a325468bdd3b756236ba.tar.gz gentoo-27e4c06d52a66978ec41a325468bdd3b756236ba.tar.bz2 gentoo-27e4c06d52a66978ec41a325468bdd3b756236ba.zip | |
sci-libs/caffe2: filtering FLASH ATTENTION by USE flash
Bug: https://bugs.gentoo.org/931046
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs')
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.3.0-r3.ebuild (renamed from sci-libs/caffe2/caffe2-2.3.0-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-2.3.0-r2.ebuild b/sci-libs/caffe2/caffe2-2.3.0-r3.ebuild index 41630a00aa19..c01e904d8eb0 100644 --- a/sci-libs/caffe2/caffe2-2.3.0-r2.ebuild +++ b/sci-libs/caffe2/caffe2-2.3.0-r3.ebuild @@ -20,7 +20,7 @@ S="${WORKDIR}"/${MYP} LICENSE="BSD" SLOT="0" KEYWORDS="~amd64" -IUSE="cuda distributed fbgemm ffmpeg gloo mkl mpi nnpack +numpy onednn openblas opencl opencv openmp qnnpack rocm xnnpack" +IUSE="cuda distributed fbgemm ffmpeg flash gloo mkl mpi nnpack +numpy onednn openblas opencl opencv openmp qnnpack rocm xnnpack" RESTRICT="test" REQUIRED_USE=" ${PYTHON_REQUIRED_USE} @@ -28,7 +28,10 @@ REQUIRED_USE=" mpi? ( distributed ) gloo? ( distributed ) ?? ( cuda rocm ) - rocm? ( || ( ${ROCM_REQUIRED_USE} ) ) + rocm? ( + || ( ${ROCM_REQUIRED_USE} ) + !flash + ) " # CUDA 12 not supported yet: https://github.com/pytorch/pytorch/issues/91122 @@ -172,6 +175,7 @@ src_configure() { -DUSE_FAKELOWP=OFF -DUSE_FBGEMM=$(usex fbgemm) -DUSE_FFMPEG=$(usex ffmpeg) + -DUSE_FLASH_ATTENTION=$(usex flash) -DUSE_GFLAGS=ON -DUSE_GLOG=ON -DUSE_GLOO=$(usex gloo) diff --git a/sci-libs/caffe2/metadata.xml b/sci-libs/caffe2/metadata.xml index ed1f9fa58993..43aee37ffc47 100644 --- a/sci-libs/caffe2/metadata.xml +++ b/sci-libs/caffe2/metadata.xml @@ -9,6 +9,7 @@ <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="flash">Enable flash attention</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> |