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 }
