summaryrefslogtreecommitdiff
path: root/sci-ml/caffe2
diff options
context:
space:
mode:
Diffstat (limited to 'sci-ml/caffe2')
-rw-r--r--sci-ml/caffe2/Manifest10
-rw-r--r--sci-ml/caffe2/caffe2-2.4.1-r6.ebuild (renamed from sci-ml/caffe2/caffe2-2.4.1-r4.ebuild)25
-rw-r--r--sci-ml/caffe2/caffe2-2.5.1-r10.ebuild (renamed from sci-ml/caffe2/caffe2-2.5.1-r8.ebuild)57
-rw-r--r--sci-ml/caffe2/caffe2-2.6.0-r2.ebuild (renamed from sci-ml/caffe2/caffe2-2.6.0.ebuild)55
-rw-r--r--sci-ml/caffe2/files/caffe2-2.5.1-newfix-functorch-install.patch2
-rw-r--r--sci-ml/caffe2/metadata.xml1
6 files changed, 83 insertions, 67 deletions
diff --git a/sci-ml/caffe2/Manifest b/sci-ml/caffe2/Manifest
index 4ddb7ca57d5a..7db834f6adf3 100644
--- a/sci-ml/caffe2/Manifest
+++ b/sci-ml/caffe2/Manifest
@@ -4,7 +4,7 @@ AUX caffe2-2.4.0-libfmt-11.patch 1560 BLAKE2B 816201b6abc3ad4c4924c649905068306f
AUX caffe2-2.5.1-cudnn_include_fix.patch 587 BLAKE2B 8ffeb9080ee77b953e7a77c9bea9af8c078adf147c314e07cf40d6f9ce1c988ac201e282ffd67a29703ced5885329d44be013c54cf3ba870c009aed40e65fefc SHA512 b37427e35d8147e603331eb344a3542ed31d0b133df3c7ce10ebecec93b1d09d040f77c33f23f70e9835db10fa209c0318b89b70b3b91263655d160ed737d6f3
AUX caffe2-2.5.1-gentoo.patch 5181 BLAKE2B 146e49ea90dbdd3bfb12a8f1b135b9ac8ed849c408a81081a5308c8dd761e1b3a8e714b905c2b88a785f436123347899b7a72eccacc2fbc54d9efad2d1dc3550 SHA512 885e79edc67d7eff1e895bdd3589704220714b6024635444dac1452e41a947c2b2a4a33866acbfc19af66d42f340038395aeb53d1477294712578e4bdbe06673
AUX caffe2-2.5.1-glog-0.6.0.patch 959 BLAKE2B eb0c18a34a2c5a1b68a420b9001eac5c8ff65ef9ca9ea1e5ad93ecef3cbaa595808105bbf6c8d3dcb9047559d23bc0398cda5922fb41e29a7588644c90c90fd8 SHA512 5873f11930ae5947c7dfbb19d0377d5535e395fcc1ab213e3c65c77b2e3bd911efc11cde2ca3e7134e9e7798ca7a483910113f331c0a23b73a5546aaaded40c5
-AUX caffe2-2.5.1-newfix-functorch-install.patch 441 BLAKE2B 0caa9e43432aecba83e3ae1cc2974d8eeb9262c8547bd1a72b0651b80a5057adb0d2bbbd2f77b773fe1036bd933dadb9f4a40e886e363f5d0a961ace69e6d16a SHA512 b8b9262db9e76768e8a114857d9bfcae67528568b5e3b90a0330dc246329439da56600689a289e9bb8aee06bd5a9708bdc2b49f79abc139aaec2778b6d68458b
+AUX caffe2-2.5.1-newfix-functorch-install.patch 455 BLAKE2B 32f1b92ce029337a6e977487b352456e7db42f9e8a51d4dbb8d3cd8bdd76f04cd9db1aead6a30ebef13f5a157e7310563e12e5af1f6e48785605e5d1e66e25ab SHA512 d78030fd4d1ee3c393a6171f0018fd101d068f321828a20fc33049ce18115dafacd5c6421c81a46c99abe46653bf862029d90df3bbbea61d90dea793432ef58c
AUX caffe2-2.5.1-unbundle_fmt.patch 499 BLAKE2B 06550f8e7ae3a932a585f8abe7653ac0b574d9ce1160c190a39f96b7d76e3a9e7a4e62b9e5e967e5884d0e38668f60af7737ea3c715d718deb7a634de98ccebd SHA512 afcb5f1123cbe5eeebdb2c5e312e37956dbc33a926cf379f531896f2c75fbb4b4dfa2c9d276a98bbad51bc587886bb572c3221be06e5116443c8ab44e1d12da5
AUX caffe2-2.5.1-unbundle_kineto.patch 830 BLAKE2B 80c9ebc30d0ca3b964141a5d9a7cb3662e6dc8472454ceabc2725f47277211cd495c4146dc9c2046cc5c5505909cf4e30c780685fceebdb05319aba5c4366346 SHA512 9645b11b5395f7a4626f17edae22c718c73d2e2f3de56e98653932a3c66f417942d54fd1dadc99112337c11513ac707a96d0caac4bf3d100db8f7079609b9b68
AUX caffe2-2.6.0-gentoo.patch 6698 BLAKE2B 58c46bedcc71ef25ab3e70efbea98e2cd4f5c068deb289ae95ba985745edcacb258bcf5cf093f44ff60606b3f6dcd1f75c9463ee225dec5e79675d0f78196e05 SHA512 b835e24e903f4d4aeaa4c1c8611bdd3896e36631c163da4154201676d7c5236e6eb7d80e0f81700f47b5f5859ced4c7ec0bdfbef6646c6ccc0f982f0d893fe14
@@ -12,7 +12,7 @@ DIST caffe2-patches-20240809.tar.gz 15242 BLAKE2B 77503c61487e7d85cca5afcab9a6e6
DIST pytorch-2.4.1.tar.gz 115029469 BLAKE2B c2909ff27d527bc57cba56b780d3b8cd07a043ab045caa6c6b27857a16f9ad10aaab2116b26226b1e46ee08ffb44007965d914464418e4ae14ca48c3f3f383bb SHA512 7e9b4485e242eaf0d648765c6621d73d95e7107b766646a098175436d1ab2e2b864badd0757a3bab6b7c318233f2120bad9ac07b39bb9e357897919580c87631
DIST pytorch-2.5.1.tar.gz 116091366 BLAKE2B 7838b17562b94ffc7d798031348689db607dd5eae2a3c35be365972e2b52a2c1b12067068d5aca5ab00cf0977d9c2c3c9ae5337d69534c864c732e6256cbeef6 SHA512 a913a466324a65fa3d79c5e9ad4d605fc7976f0134fda2f81aaa3cea29d56926604999b8a238759646d211e63b47bbb446cdffa86ca8defd8159f11e30301289
DIST pytorch-2.6.0.tar.gz 119594438 BLAKE2B 3152eb341cf42295e147e59625beb9c06608aa4b78f9618c1c0024b10c1c767715d07fe8c4be52d029ac47f808cd0d5e65c9530ec90d951a64b993083b4067ad SHA512 a70da80ff09d226085e18228132cf6bb236ad8cc47eed52375d0d2a615f09dd33849da947270b5670c184eab60cb8e2adf11d801babfbda7aa621400501d07b0
-EBUILD caffe2-2.4.1-r4.ebuild 8493 BLAKE2B 0e87ccde293ada0765a76580c82a8a9011eae0e554765347b2911f660587895758af79d774439f630806c4644f0e7f74f256b3efc3a6de94d049e46d0031d71b SHA512 28ee9b684719bace598d99f2fa532f91fe7e33882a2f2df4ba649fb000242c4bd09cb360c887da04e22c099c501effef3adab6f3fc299291e821888e797e4dc2
-EBUILD caffe2-2.5.1-r8.ebuild 8658 BLAKE2B 2cbc6401ad7a276a415feb2bb3652ce758201fff9252333384ef32aef21d91da01365f26ba46757f66d8b82d8d95ef223bb5807e111d308f241de013946c1376 SHA512 7f687014d47eac625e3496b46d29ff4dffbc3c4d9e0425b545b6b9495b2e9e80f5f3fa2ea42d2e0ff3f01e2555ead21f432736e0365caa34d1aed108a83d436f
-EBUILD caffe2-2.6.0.ebuild 8681 BLAKE2B 53ef41565f538c171e063a76fac06bfe5e8c4ac5c9ced917fbc07f359dc4e51feb70e182b7008254e18fdb9df4bb33df10442e8392efd4ca52f30da61f25f005 SHA512 0e909d6227b4f8bc2b0f7900bada6fd9bdd2bd410c7c64ef81a0c7fb2c8a2742e679f3e67946fdb6276f3246a817f5279604cd196e802124efb2bbeda0a589e6
-MISC metadata.xml 1162 BLAKE2B 8dba3cb46f3a2ec3be3e5b6fec7d6702095d926bde0416543f596021515775a3a870e1a3a0bb52e3b0e6b64254eef5ac963d42b678cd91f6b88377d292d4de50 SHA512 847ee1c39d29cf46d1a0b2c39563048aefc64daed00db50790206947bbf3e35cad4b52a15f92e760c5143a466283e33e1cc7e4bd70e1b48be584c805b2365b81
+EBUILD caffe2-2.4.1-r6.ebuild 8556 BLAKE2B 77f1809c8ec73b36e16a8c2939bcafb83a995f8074c163e224f0b02230f6c9cde73c2c11aa07b44f4b670b4472c6ba4157262dd14743cad385187bc4145f5af9 SHA512 7747b9ad89f065af36e5d191f24525748a4c6ea020848b0d64f5699e8f9192699ec523708ea6bc52e1973c5792e65330ba7d27e3a0c88d0990094feda9e9a46d
+EBUILD caffe2-2.5.1-r10.ebuild 8866 BLAKE2B bcc3af372f8ab5969903c32bf15cc619d700776133cfa20e723fe126435505e73fcf48985c338c3055b5a206f776fe5744a73441308114356f35e566aae56b28 SHA512 e175f3741c80e0181ec03952dfac7bcfad808275c60d01f18a46073300d575b31f7b161eb715db22a00e8bb13e37c68051e487982a8f020d63da342da24ff56e
+EBUILD caffe2-2.6.0-r2.ebuild 8889 BLAKE2B 0ccf998a778c69efb005967a5ed6110c34ab263e42f5d969fb4bbbed611209ae1c9191df8c47b53c7c194b2612ce86229456dc4fc6383779a8736e4499056bfa SHA512 d5f19c8f8742b0bfacef9656b9ead6bdaf1185360467d6b83cd1bc7e26db2e3e2c925131850a7f2b9bf3b1fadc081e9d6047ae3b86aeacecc689377fe62edca8
+MISC metadata.xml 1228 BLAKE2B 29fa99ccc009ae8ac209cf0a85a8c87992d10301837f44478b4fe60445dbf261cf1c3869cc6e173944b9c98608fcf32a379c766bf7e53267c90fd71e51e1f9a0 SHA512 06d89ac1b62bf215d56ac67408f6027849f3a1948f1a40013aa989f9d879eb9aa9156d9c336cc83d2bb3795271c11d39d34919f749d83dbfbf48cc2d6aa8545e
diff --git a/sci-ml/caffe2/caffe2-2.4.1-r4.ebuild b/sci-ml/caffe2/caffe2-2.4.1-r6.ebuild
index 9fb7f21f1712..bc95abbd3fca 100644
--- a/sci-ml/caffe2/caffe2-2.4.1-r4.ebuild
+++ b/sci-ml/caffe2/caffe2-2.4.1-r6.ebuild
@@ -3,7 +3,7 @@
EAPI=8
-PYTHON_COMPAT=( python3_{10..12} )
+PYTHON_COMPAT=( python3_{10..13} )
ROCM_VERSION=6.1
inherit python-single-r1 cmake cuda flag-o-matic prefix rocm toolchain-funcs
@@ -67,7 +67,7 @@ RDEPEND="
opencl? ( virtual/opencl )
qnnpack? (
!sci-libs/QNNPACK
- dev-cpp/gemmlowp
+ sci-ml/gemmlowp
)
rocm? (
=dev-util/hip-6.1*
@@ -100,19 +100,19 @@ RDEPEND="
DEPEND="
${RDEPEND}
dev-libs/clog
- cuda? ( <=dev-libs/cutlass-3.4.1 )
- onednn? ( sci-ml/ideep )
dev-libs/psimd
- sci-ml/FP16
dev-libs/FXdiv
dev-libs/pocketfft
dev-libs/flatbuffers
+ sci-ml/FP16
sci-ml/kineto
$(python_gen_cond_dep '
- dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/pybind11[${PYTHON_USEDEP}]
+ dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/typing-extensions[${PYTHON_USEDEP}]
')
+ cuda? ( <=dev-libs/cutlass-3.4.1 )
+ onednn? ( sci-ml/ideep )
"
PATCHES=(
@@ -283,9 +283,17 @@ src_configure() {
sed '/RERUN/,+1d' -i "${BUILD_DIR}"/build.ninja || die
}
+python_install() {
+ python_domodule python/caffe2
+ python_domodule python/torch
+ ln -s ../../../../../include/torch \
+ "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
+}
+
src_install() {
cmake_src_install
+ # Used by pytorch ebuild
insinto "/var/lib/${PN}"
doins "${BUILD_DIR}"/CMakeCache.txt
@@ -293,8 +301,5 @@ src_install() {
mkdir -p python/torch/include || die
mv "${ED}"/usr/lib/python*/site-packages/caffe2 python/ || die
cp torch/version.py python/torch/ || die
- python_domodule python/caffe2
- python_domodule python/torch
- ln -s ../../../../../include/torch \
- "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
+ python_install
}
diff --git a/sci-ml/caffe2/caffe2-2.5.1-r8.ebuild b/sci-ml/caffe2/caffe2-2.5.1-r10.ebuild
index 848c743feb57..02c57f637949 100644
--- a/sci-ml/caffe2/caffe2-2.5.1-r8.ebuild
+++ b/sci-ml/caffe2/caffe2-2.5.1-r10.ebuild
@@ -3,7 +3,7 @@
EAPI=8
-PYTHON_COMPAT=( python3_{10..12} )
+PYTHON_COMPAT=( python3_{10..13} )
ROCM_VERSION=6.1
inherit python-single-r1 cmake cuda flag-o-matic prefix rocm toolchain-funcs
@@ -20,7 +20,8 @@ S="${WORKDIR}"/${MYP}
LICENSE="BSD"
SLOT="0"
KEYWORDS="~amd64"
-IUSE="cuda distributed fbgemm flash gloo mkl mpi nnpack +numpy onednn openblas opencl openmp qnnpack rocm xnnpack"
+IUSE="cuda distributed fbgemm flash gloo memefficient mkl mpi nnpack +numpy
+ onednn openblas opencl openmp qnnpack rocm xnnpack"
RESTRICT="test"
REQUIRED_USE="
${PYTHON_REQUIRED_USE}
@@ -45,9 +46,9 @@ RDEPEND="
dev-libs/protobuf:=
dev-libs/pthreadpool
dev-libs/sleef
- virtual/lapack
sci-ml/onnx
sci-ml/foxi
+ virtual/lapack
cuda? (
dev-libs/cudnn
>=dev-libs/cudnn-frontend-1.0.3:0/8
@@ -64,7 +65,7 @@ RDEPEND="
opencl? ( virtual/opencl )
qnnpack? (
!sci-libs/QNNPACK
- dev-cpp/gemmlowp
+ sci-ml/gemmlowp
)
rocm? (
=dev-util/hip-6.1*
@@ -97,20 +98,20 @@ RDEPEND="
DEPEND="
${RDEPEND}
- qnnpack? ( dev-libs/clog )
- cuda? ( <=dev-libs/cutlass-3.4.1 )
- onednn? ( sci-ml/ideep )
- dev-libs/psimd
- sci-ml/FP16
+ dev-libs/flatbuffers
dev-libs/FXdiv
dev-libs/pocketfft
- dev-libs/flatbuffers
+ dev-libs/psimd
+ sci-ml/FP16
sci-ml/kineto
$(python_gen_cond_dep '
- dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/pybind11[${PYTHON_USEDEP}]
+ dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/typing-extensions[${PYTHON_USEDEP}]
')
+ cuda? ( <=dev-libs/cutlass-3.4.1 )
+ onednn? ( sci-ml/ideep )
+ qnnpack? ( dev-libs/clog )
"
PATCHES=(
@@ -149,6 +150,10 @@ src_prepare() {
c10/CMakeLists.txt \
c10/hip/CMakeLists.txt \
|| die
+ sed -i \
+ -e '/Using pocketfft in directory:/d' \
+ cmake/Dependencies.cmake \
+ || die
cmake_src_prepare
pushd torch/csrc/jit/serialization || die
@@ -213,7 +218,7 @@ src_configure() {
-DUSE_ITT=OFF
-DUSE_KINETO=OFF # TODO
-DUSE_MAGMA=OFF # TODO: In GURU as sci-libs/magma
- -DUSE_MEM_EFF_ATTENTION=OFF
+ -DUSE_MEM_EFF_ATTENTION=$(usex memefficient)
-DUSE_MKLDNN=$(usex onednn)
-DUSE_MPI=$(usex mpi)
-DUSE_NCCL=OFF
@@ -298,6 +303,19 @@ src_compile() {
cmake_src_compile
}
+python_install() {
+ python_domodule python/torch
+ mkdir "${D}"$(python_get_sitedir)/torch/bin || die
+ mkdir "${D}"$(python_get_sitedir)/torch/lib || die
+ mkdir "${D}"$(python_get_sitedir)/torch/include || die
+ ln -s ../../../../../include/torch \
+ "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
+ ln -s ../../../../../bin/torch_shm_manager \
+ "${D}"/$(python_get_sitedir)/torch/bin/torch_shm_manager || die
+ ln -s ../../../../../$(get_libdir)/libtorch_global_deps.so \
+ "${D}"/$(python_get_sitedir)/torch/lib/libtorch_global_deps.so || die
+}
+
src_install() {
cmake_src_install
@@ -309,18 +327,5 @@ src_install() {
rm -rf python
mkdir -p python/torch || die
cp torch/version.py python/torch/ || die
- python_domodule python/torch
-
- mkdir "${D}"$(python_get_sitedir)/torch/bin || die
- mkdir "${D}"$(python_get_sitedir)/torch/lib || die
- mkdir "${D}"$(python_get_sitedir)/torch/include || die
-
- ln -s ../../../../../include/torch \
- "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
-
- mv "${D}"/usr/bin/torch_shm_manager \
- "${D}"/$(python_get_sitedir)/torch/bin/ || die
-
- mv "${D}"/usr/$(get_libdir)/libtorch_global_deps.so \
- "${D}"/$(python_get_sitedir)/torch/lib/ || die
+ python_install
}
diff --git a/sci-ml/caffe2/caffe2-2.6.0.ebuild b/sci-ml/caffe2/caffe2-2.6.0-r2.ebuild
index b7d62a0ae814..b9747ad9eed4 100644
--- a/sci-ml/caffe2/caffe2-2.6.0.ebuild
+++ b/sci-ml/caffe2/caffe2-2.6.0-r2.ebuild
@@ -20,7 +20,8 @@ S="${WORKDIR}"/${MYP}
LICENSE="BSD"
SLOT="0"
KEYWORDS="~amd64"
-IUSE="cuda distributed fbgemm flash gloo mkl mpi nnpack +numpy onednn openblas opencl openmp qnnpack rocm xnnpack"
+IUSE="cuda distributed fbgemm flash gloo memefficient mkl mpi nnpack +numpy
+ onednn openblas opencl openmp qnnpack rocm xnnpack"
RESTRICT="test"
REQUIRED_USE="
${PYTHON_REQUIRED_USE}
@@ -46,8 +47,8 @@ RDEPEND="
dev-libs/pthreadpool
dev-libs/sleef
virtual/lapack
- sci-ml/onnx
sci-ml/foxi
+ sci-ml/onnx
cuda? (
dev-libs/cudnn
>=dev-libs/cudnn-frontend-1.0.3:0/8
@@ -64,7 +65,7 @@ RDEPEND="
opencl? ( virtual/opencl )
qnnpack? (
!sci-libs/QNNPACK
- dev-cpp/gemmlowp
+ sci-ml/gemmlowp
)
rocm? (
=dev-util/hip-6.1*
@@ -97,20 +98,20 @@ RDEPEND="
DEPEND="
${RDEPEND}
- qnnpack? ( dev-libs/clog )
- cuda? ( <=dev-libs/cutlass-3.4.1 )
- onednn? ( sci-ml/ideep )
- dev-libs/psimd
- sci-ml/FP16
+ dev-libs/flatbuffers
dev-libs/FXdiv
dev-libs/pocketfft
- dev-libs/flatbuffers
+ dev-libs/psimd
+ sci-ml/FP16
sci-ml/kineto
$(python_gen_cond_dep '
- dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/pybind11[${PYTHON_USEDEP}]
+ dev-python/pyyaml[${PYTHON_USEDEP}]
dev-python/typing-extensions[${PYTHON_USEDEP}]
')
+ cuda? ( <=dev-libs/cutlass-3.4.1 )
+ onednn? ( sci-ml/ideep )
+ qnnpack? ( dev-libs/clog )
"
PATCHES=(
@@ -149,6 +150,10 @@ src_prepare() {
c10/CMakeLists.txt \
c10/hip/CMakeLists.txt \
|| die
+ sed -i \
+ -e '/Using pocketfft in directory:/d' \
+ cmake/Dependencies.cmake \
+ || die
cmake_src_prepare
pushd torch/csrc/jit/serialization || die
@@ -213,7 +218,7 @@ src_configure() {
-DUSE_ITT=OFF
-DUSE_KINETO=OFF # TODO
-DUSE_MAGMA=OFF # TODO: In GURU as sci-libs/magma
- -DUSE_MEM_EFF_ATTENTION=OFF
+ -DUSE_MEM_EFF_ATTENTION=$(usex memefficient)
-DUSE_MKLDNN=$(usex onednn)
-DUSE_MPI=$(usex mpi)
-DUSE_NCCL=OFF
@@ -298,6 +303,19 @@ src_compile() {
cmake_src_compile
}
+python_install() {
+ python_domodule python/torch
+ mkdir "${D}"$(python_get_sitedir)/torch/bin || die
+ mkdir "${D}"$(python_get_sitedir)/torch/lib || die
+ mkdir "${D}"$(python_get_sitedir)/torch/include || die
+ ln -s ../../../../../include/torch \
+ "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
+ ln -s ../../../../../bin/torch_shm_manager \
+ "${D}"/$(python_get_sitedir)/torch/bin/torch_shm_manager || die
+ ln -s ../../../../../$(get_libdir)/libtorch_global_deps.so \
+ "${D}"/$(python_get_sitedir)/torch/lib/libtorch_global_deps.so || die
+}
+
src_install() {
cmake_src_install
@@ -309,18 +327,5 @@ src_install() {
rm -rf python
mkdir -p python/torch || die
cp torch/version.py python/torch/ || die
- python_domodule python/torch
-
- mkdir "${D}"$(python_get_sitedir)/torch/bin || die
- mkdir "${D}"$(python_get_sitedir)/torch/lib || die
- mkdir "${D}"$(python_get_sitedir)/torch/include || die
-
- ln -s ../../../../../include/torch \
- "${D}$(python_get_sitedir)"/torch/include/torch || die # bug 923269
-
- mv "${D}"/usr/bin/torch_shm_manager \
- "${D}"/$(python_get_sitedir)/torch/bin/ || die
-
- mv "${D}"/usr/$(get_libdir)/libtorch_global_deps.so \
- "${D}"/$(python_get_sitedir)/torch/lib/ || die
+ python_install
}
diff --git a/sci-ml/caffe2/files/caffe2-2.5.1-newfix-functorch-install.patch b/sci-ml/caffe2/files/caffe2-2.5.1-newfix-functorch-install.patch
index 499a3c74c87f..91eed46b813e 100644
--- a/sci-ml/caffe2/files/caffe2-2.5.1-newfix-functorch-install.patch
+++ b/sci-ml/caffe2/files/caffe2-2.5.1-newfix-functorch-install.patch
@@ -5,4 +5,4 @@
set_target_properties(${PROJECT_NAME} PROPERTIES LINK_FLAGS ${TORCH_PYTHON_LINK_FLAGS})
endif()
-install(TARGETS ${PROJECT_NAME} DESTINATION "${CMAKE_CURRENT_SOURCE_DIR}")
-+install(TARGETS ${PROJECT_NAME} DESTINATION "/var/lib/caffe2/")
++install(TARGETS ${PROJECT_NAME} DESTINATION "$ENV{$EPREFIX}/var/lib/caffe2/")
diff --git a/sci-ml/caffe2/metadata.xml b/sci-ml/caffe2/metadata.xml
index 0754850fe24c..2483e58ed39e 100644
--- a/sci-ml/caffe2/metadata.xml
+++ b/sci-ml/caffe2/metadata.xml
@@ -10,6 +10,7 @@
<flag name="fbgemm">Use <pkg>sci-ml/FBGEMM</pkg></flag>
<flag name="flash">Enable flash attention</flag>
<flag name="gloo">Use <pkg>sci-ml/gloo</pkg></flag>
+ <flag name="memefficient">Enable mem efficient attention</flag>
<flag name="mkl">Use <pkg>sci-libs/mkl</pkg> for blas, lapack and sparse blas routines</flag>
<flag name="nnpack">Use <pkg>sci-ml/NNPACK</pkg></flag>
<flag name="numpy">Add support for math operations through numpy</flag>
generated by cgit v1.2.3 (git 2.25.1) at 2025-04-10 00:44:13 +0000