summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAlfredo Tupone <tupone@gentoo.org>2023-12-14 19:29:37 +0100
committerAlfredo Tupone <tupone@gentoo.org>2023-12-14 19:30:24 +0100
commit52575e06cb70d8f89dfbcfa3b40c4b216b4f35db (patch)
tree0587f18292b002af2deccf9db25b608dcbc0c4ba /sci-libs
parentdev-python/trove-classifiers: Use flit-core, for hatchling bootstrap (diff)
downloadgentoo-52575e06cb70d8f89dfbcfa3b40c4b216b4f35db.tar.gz
gentoo-52575e06cb70d8f89dfbcfa3b40c4b216b4f35db.tar.bz2
gentoo-52575e06cb70d8f89dfbcfa3b40c4b216b4f35db.zip
sci-libs/caffe2: adj deps
Closes: https://bugs.gentoo.org/919744 Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs')
-rw-r--r--sci-libs/caffe2/caffe2-2.1.1-r2.ebuild (renamed from sci-libs/caffe2/caffe2-2.1.1-r1.ebuild)10
1 files changed, 6 insertions, 4 deletions
diff --git a/sci-libs/caffe2/caffe2-2.1.1-r1.ebuild b/sci-libs/caffe2/caffe2-2.1.1-r2.ebuild
index f5aafac33607..714b5477c38f 100644
--- a/sci-libs/caffe2/caffe2-2.1.1-r1.ebuild
+++ b/sci-libs/caffe2/caffe2-2.1.1-r2.ebuild
@@ -43,10 +43,10 @@ RDEPEND="
sci-libs/foxi
cuda? (
=dev-libs/cudnn-8*
- dev-libs/cudnn-frontend:0/8
+ >=dev-libs/cudnn-frontend-0.9.2:0/8
<dev-util/nvidia-cuda-toolkit-12:=[profiler]
)
- fbgemm? ( dev-libs/FBGEMM )
+ fbgemm? ( >=dev-libs/FBGEMM-2023.11.02 )
ffmpeg? ( media-video/ffmpeg:= )
gloo? ( sci-libs/gloo[cuda?] )
mpi? ( virtual/mpi )
@@ -63,7 +63,7 @@ RDEPEND="
DEPEND="
${RDEPEND}
dev-cpp/eigen
- cuda? ( dev-libs/cutlass )
+ cuda? ( >=dev-libs/cutlass-3.1.0 )
dev-libs/psimd
dev-libs/FP16
dev-libs/FXdiv
@@ -203,7 +203,9 @@ src_install() {
mkdir -p python/torch/include || die
mv "${ED}"/usr/lib/python*/site-packages/caffe2 python/ || die
mv "${ED}"/usr/include/torch python/torch/include || die
- mv "${ED}${S}"/nvfuser python/nvfuser || die
+ if use cuda; then
+ mv "${ED}${S}"/nvfuser python/nvfuser || die
+ fi
rm -r "${ED}${S}"/test || die
rm -r "${ED}${S}"/third_party || die
cp torch/version.py python/torch/ || die