diff options
| author |   Alfredo Tupone <tupone@gentoo.org> | 2023-12-06 20:48:46 +0100 |
|---|---|---|
| committer | Alfredo Tupone <tupone@gentoo.org> | 2023-12-06 20:49:16 +0100 |
| commit | a30ecc1b69628e20faa989943ff1b0dda32d9d69 (patch) | |
| tree | f0900b31ca226f35e556471ae62500d156889c8f /sci-libs/caffe2 | |
| parent | net-misc/freerdp: fix build with clang (diff) | |
| download | gentoo-a30ecc1b69628e20faa989943ff1b0dda32d9d69.tar.gz gentoo-a30ecc1b69628e20faa989943ff1b0dda32d9d69.tar.bz2 gentoo-a30ecc1b69628e20faa989943ff1b0dda32d9d69.zip | |
sci-libs/caffe2: install nvfuser python module
Bug: https://bugs.gentoo.org/914572
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs/caffe2')
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.1.1.ebuild | 6 |
1 files changed, 5 insertions, 1 deletions
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 } |