commit: 39dacf4a61719bf8d19014b94bafd88c6f5b57d5 Author: Alfredo Tupone <tupone <AT> gentoo <DOT> org> AuthorDate: Wed Jul 2 09:02:07 2025 +0000 Commit: Alfredo Tupone <tupone <AT> gentoo <DOT> org> CommitDate: Wed Jul 2 09:03:11 2025 +0000 URL: https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=39dacf4a
sci-ml/caffe2: update SRC_URI Bug: https://bugs.gentoo.org/956674 Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org> sci-ml/caffe2/Manifest | 1 + sci-ml/caffe2/caffe2-2.7.1.ebuild | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) diff --git a/sci-ml/caffe2/Manifest b/sci-ml/caffe2/Manifest index fda2a4a29e01..9510577bc397 100644 --- a/sci-ml/caffe2/Manifest +++ b/sci-ml/caffe2/Manifest @@ -1,3 +1,4 @@ +DIST aotriton-0.9.2b-manylinux_2_28_x86_64-rocm6.3-shared.tar.gz 444786966 BLAKE2B 38ebf7edd1686d137bf70022a50ce2b7c060eb8420fe11e406074d3531e84538f4aeb198f1aafe34abf3f5b1262f3edd6e81d204eb84ec320f8ba8180387313b SHA512 b0d6d25ae4be9272d43bd80fa9d0178b27f5feb1b83bfed50e87c7efedfdd66a18981d0f3fabf8087b1c476d9937eaa80d462dea26502b24702134145bd09394 DIST composable_kernel-50ee4267.tar.gz 4194795 BLAKE2B b3c97d98a0c9e4620fdae3d30006edf55cc60ffa7f8518f6acb8d808647bc4de362c2e2b7e974686503fa2c7f359b6981cfbda74e40cc1bad4d351c5d2ff92e1 SHA512 9fc6f5f15556f020414b4567520329ef762209a82411a246c2bc1240a9fed2669f7fcb982cf773e3e9561bf9a2c557dba82b8b469d2e5844e679e2f5ab7c3e17 DIST composable_kernel-8086bbe3.tar.gz 4418862 BLAKE2B b710e3d4586899443ec01044dad19fd2f992c351e2f65ba526dfcc47cc65c095beaf8ac21a8f71c02a0eb524d364e817b27241a9198884f2bdae9924b51e24e4 SHA512 8410b5a1c864d71f3034ef0d9d1245078856d09cc191faec59856c229bf11d89ae291036d735cb5cec4f1d72e6e9e8f6921833147f9619d30cfab8722d3a9f63 DIST flash-attention-2.7.4.gh.tar.gz 5841323 BLAKE2B 432999d763f2b3d732580ddfea5d3e01370351db0656546259a5e500a07516dd03c98828bfb55855dabe4adc651033b5d97ea4725ca46158b9970f0fbc662710 SHA512 05a4afb09e666f7404d6a3f8b5256e7bed6eba60a6f1bde2b7dbb96d318975f0b458c2521c7a38d88e97b6e4c27f29077cf787849daf82586e33f43a3d9a84b3 diff --git a/sci-ml/caffe2/caffe2-2.7.1.ebuild b/sci-ml/caffe2/caffe2-2.7.1.ebuild index 9a8498a37554..9a4c0fbb3f55 100644 --- a/sci-ml/caffe2/caffe2-2.7.1.ebuild +++ b/sci-ml/caffe2/caffe2-2.7.1.ebuild @@ -19,6 +19,11 @@ FLASH_PV=2.7.4 FLASH_PN=flash-attention FLASH_P=${FLASH_PN}-${FLASH_PV} +AOTRITON_PV=0.9.2b +AOTRITON_PN=aotriton +AOTRITON_P=${AOTRITON_PN}-${AOTRITON_PV} +AOTRITON_tar=${AOTRITON_P}-manylinux_2_28_x86_64-rocm6.3-shared.tar.gz + DESCRIPTION="A deep learning framework" HOMEPAGE="https://pytorch.org/"; SRC_URI=" @@ -26,6 +31,11 @@ SRC_URI=" rocm? ( https://github.com/ROCm/composable_kernel/archive/${CK_COMMIT}.tar.gz -> ${CK_P}.tar.gz + memefficient? ( + amd64? ( + https://github.com/ROCm/${AOTRITON_PN}/releases/download/${AOTRITON_PV}/${AOTRITON_tar} + ) + ) ) flash? ( https://github.com/Dao-AILab/${FLASH_PN}/archive/refs/tags/v${FLASH_PV}.tar.gz @@ -183,6 +193,13 @@ src_prepare() { pushd torch/csrc/jit/serialization || die flatc --cpp --gen-mutable --scoped-enums mobile_bytecode.fbs || die popd + if use rocm && use memefficient; then + mkdir -p "${BUILD_DIR}"/aotriton_external-prefix/src || die + rm -rf "${WORKDIR}"/aotriton + if use amd64; then + cp "${DISTDIR}"/${AOTRITON_tar} "${BUILD_DIR}"/aotriton_external-prefix/src || die + fi + fi # prefixify the hardcoded paths, after all patches are applied hprefixify \
