diff options
Diffstat (limited to 'sci-ml/caffe2')
| -rw-r--r-- | sci-ml/caffe2/Manifest | 6 | ||||
| -rw-r--r-- | sci-ml/caffe2/caffe2-2.5.1-r8.ebuild | 13 | ||||
| -rw-r--r-- | sci-ml/caffe2/caffe2-2.6.0.ebuild | 13 | ||||
| -rw-r--r-- | sci-ml/caffe2/metadata.xml | 1 |
4 files changed, 26 insertions, 7 deletions
diff --git a/sci-ml/caffe2/Manifest b/sci-ml/caffe2/Manifest index 4ddb7ca57d5a..d6abebf5c38e 100644 --- a/sci-ml/caffe2/Manifest +++ b/sci-ml/caffe2/Manifest @@ -13,6 +13,6 @@ DIST pytorch-2.4.1.tar.gz 115029469 BLAKE2B c2909ff27d527bc57cba56b780d3b8cd07a0 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.5.1-r8.ebuild 8902 BLAKE2B db406a189c77b077fd6bb331284ef8b98f5d24b750984e463f602ce65e6c29fe9d9de24d6a4aa01f8096c683d34d82a024fd90f470c15cab347535d5fb9def08 SHA512 959b42ba518e247571caa6569f21a89e2125b8d407120bcc0a7c37fc4832115bcf13496c86b454745d4ea4352c4ba9938ab5b31f460804c28ee8ab65cb6761e0 +EBUILD caffe2-2.6.0.ebuild 8925 BLAKE2B 375aee7e8530bb662910afd62aaa07b629c6a0d78079cd13ae7f5630d90db2dda5fbf1d9e09837df5745d77b5fc06f1bfc74daf2e5eaeb8d81baab3936e64e95 SHA512 fa0e70ef18600cf84832f7b20f63aa5671c1861406c70ecdb8f5f6fe50897e88be47694e6c7a5e7de621238d5b0b26d1d6240ae24d8ccf9dc8222d94b44e8b1b +MISC metadata.xml 1228 BLAKE2B 29fa99ccc009ae8ac209cf0a85a8c87992d10301837f44478b4fe60445dbf261cf1c3869cc6e173944b9c98608fcf32a379c766bf7e53267c90fd71e51e1f9a0 SHA512 06d89ac1b62bf215d56ac67408f6027849f3a1948f1a40013aa989f9d879eb9aa9156d9c336cc83d2bb3795271c11d39d34919f749d83dbfbf48cc2d6aa8545e diff --git a/sci-ml/caffe2/caffe2-2.5.1-r8.ebuild b/sci-ml/caffe2/caffe2-2.5.1-r8.ebuild index 848c743feb57..902bee88e3f5 100644 --- a/sci-ml/caffe2/caffe2-2.5.1-r8.ebuild +++ b/sci-ml/caffe2/caffe2-2.5.1-r8.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} @@ -149,6 +150,14 @@ src_prepare() { c10/CMakeLists.txt \ c10/hip/CMakeLists.txt \ || die + sed -i \ + -e 's:pocketfft_hdronly.h:pocketfft/pocketfft_hdronly.h:' \ + aten/src/ATen/native/mkl/SpectralOps.cpp \ + die + sed -i \ + -e '/Using pocketfft in directory:/d' \ + cmake/Dependencies.cmake \ + || die cmake_src_prepare pushd torch/csrc/jit/serialization || die @@ -213,7 +222,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 diff --git a/sci-ml/caffe2/caffe2-2.6.0.ebuild b/sci-ml/caffe2/caffe2-2.6.0.ebuild index b7d62a0ae814..1cd499d32116 100644 --- a/sci-ml/caffe2/caffe2-2.6.0.ebuild +++ b/sci-ml/caffe2/caffe2-2.6.0.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} @@ -149,6 +150,14 @@ src_prepare() { c10/CMakeLists.txt \ c10/hip/CMakeLists.txt \ || die + sed -i \ + -e 's:pocketfft_hdronly.h:pocketfft/pocketfft_hdronly.h:' \ + aten/src/ATen/native/mkl/SpectralOps.cpp \ + die + sed -i \ + -e '/Using pocketfft in directory:/d' \ + cmake/Dependencies.cmake \ + || die cmake_src_prepare pushd torch/csrc/jit/serialization || die @@ -213,7 +222,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 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> |