summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'sci-libs')
-rw-r--r--sci-libs/caffe2/caffe2-2.1.1-r6.ebuild (renamed from sci-libs/caffe2/caffe2-2.1.1-r5.ebuild)1
1 files changed, 1 insertions, 0 deletions
diff --git a/sci-libs/caffe2/caffe2-2.1.1-r5.ebuild b/sci-libs/caffe2/caffe2-2.1.1-r6.ebuild
index 1dc4f2a32587..b22e5e52ead5 100644
--- a/sci-libs/caffe2/caffe2-2.1.1-r5.ebuild
+++ b/sci-libs/caffe2/caffe2-2.1.1-r6.ebuild
@@ -205,6 +205,7 @@ src_install() {
mv "${ED}"/usr/include/torch python/torch/include || die
if use cuda; then
mv "${ED}${S}"/nvfuser python/nvfuser || die
+ mv "${ED}"/usr/$(get_libdir)/nvfuser.so python/nvfuser/_C.so || die
fi
rm -rf "${ED}${S}"/test
rm -rf "${ED}${S}"/third_party