[gentoo-commits] repo/gentoo:master commit in: sci-libs/caffe2/

["Alfredo Tupone" <tupone@gentoo.org>]

commit:     a30ecc1b69628e20faa989943ff1b0dda32d9d69
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Wed Dec  6 19:48:46 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Wed Dec  6 19:49:16 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=a30ecc1b

sci-libs/caffe2: install nvfuser python module

Bug: https://bugs.gentoo.org/914572
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-libs/caffe2/caffe2-2.1.1.ebuild | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/sci-libs/caffe2/caffe2-2.1.1.ebuild b/sci-libs/caffe2/caffe2-2.1.1.ebuild
index e5e9a71069cd..65c5c2831826 100644
--- a/sci-libs/caffe2/caffe2-2.1.1.ebuild
+++ b/sci-libs/caffe2/caffe2-2.1.1.ebuild
@@ -135,7 +135,6 @@ src_configure() {
 		-DUSE_CCACHE=OFF
 		-DUSE_CUDA=$(usex cuda)
 		-DUSE_CUDNN=$(usex cuda)
-		-DUSE_FAST_NVCC=$(usex cuda)
 		-DTORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5 7.0}"
 		-DBUILD_NVFUSER=$(usex cuda)
 		-DUSE_DISTRIBUTED=$(usex distributed)
@@ -204,7 +203,12 @@ 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
+	rm -r "${ED}${S}"/test || die
+	rm -r "${ED}${S}"/third_party || die
 	cp torch/version.py python/torch/ || die
 	python_domodule python/caffe2
 	python_domodule python/torch
+	python_domodule python/nvfuser
+	find "${ED}" -empty -delete
 }