commit: 2728535b0cef51e61a3943b822b49ed88973973d
Author: Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Fri Sep 5 17:15:32 2025 +0000
Commit: Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Fri Sep 5 17:16:07 2025 +0000
URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=2728535b
sci-ml/caffe2: memefficient require flash-attention files
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>
sci-ml/caffe2/{caffe2-2.8.0.ebuild => caffe2-2.8.0-r1.ebuild} | 9 ++++-----
1 file changed, 4 insertions(+), 5 deletions(-)
diff --git a/sci-ml/caffe2/caffe2-2.8.0.ebuild
b/sci-ml/caffe2/caffe2-2.8.0-r1.ebuild
similarity index 97%
rename from sci-ml/caffe2/caffe2-2.8.0.ebuild
rename to sci-ml/caffe2/caffe2-2.8.0-r1.ebuild
index 161235c2c32c..75e22030d96f 100644
--- a/sci-ml/caffe2/caffe2-2.8.0.ebuild
+++ b/sci-ml/caffe2/caffe2-2.8.0-r1.ebuild
@@ -18,6 +18,7 @@ CK_P=composable_kernel-${CK_COMMIT:0:8}
FLASH_PV=2.7.4
FLASH_PN=flash-attention
FLASH_P=${FLASH_PN}-${FLASH_PV}
+FLASH_ATT_URI="https://github.com/Dao-AILab/${FLASH_PN}/archive/refs/tags/v${FLASH_PV}.tar.gz
-> ${FLASH_P}.gh.tar.gz"
AOTRITON_PV=0.9.2b
AOTRITON_PN=aotriton
@@ -32,10 +33,8 @@ SRC_URI="
https://github.com/ROCm/composable_kernel/archive/${CK_COMMIT}.tar.gz
-> ${CK_P}.tar.gz
)
- flash? (
-
https://github.com/Dao-AILab/${FLASH_PN}/archive/refs/tags/v${FLASH_PV}.tar.gz
- -> ${FLASH_P}.gh.tar.gz
- )
+ flash? ( ${FLASH_ATT_URI} )
+ memefficient? ( ${FLASH_ATT_URI} )
"
S="${WORKDIR}"/${MYP}
@@ -151,7 +150,7 @@ PATCHES=(
)
src_prepare() {
- if use flash; then
+ if use flash || use memefficient; then
mv "${WORKDIR}"/${FLASH_P}/* third_party/${FLASH_PN}/ || die
fi
filter-lto #bug 862672
