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

[Alfredo Tupone]

commit:     5088a2d38c87926670b7d18f75fc6b91da982caf
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Fri Aug 11 07:26:35 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Fri Aug 11 07:31:10 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=5088a2d3

sci-libs/caffe2: build nvfuser for cuda users

Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-libs/caffe2/{caffe2-2.0.0-r4.ebuild => caffe2-2.0.0-r5.ebuild} | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/sci-libs/caffe2/caffe2-2.0.0-r4.ebuild 
b/sci-libs/caffe2/caffe2-2.0.0-r5.ebuild
similarity index 98%
rename from sci-libs/caffe2/caffe2-2.0.0-r4.ebuild
rename to sci-libs/caffe2/caffe2-2.0.0-r5.ebuild
index f6a3684a96a0..eaf3e07d46ac 100644
--- a/sci-libs/caffe2/caffe2-2.0.0-r4.ebuild
+++ b/sci-libs/caffe2/caffe2-2.0.0-r5.ebuild
@@ -121,7 +121,7 @@ src_configure() {
                -DUSE_CUDNN=$(usex cuda)
                -DUSE_FAST_NVCC=$(usex cuda)
                -DTORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5 7.0}"
-               -DBUILD_NVFUSER=OFF
+               -DBUILD_NVFUSER=$(usex cuda)
                -DUSE_DISTRIBUTED=$(usex distributed)
                -DUSE_MPI=$(usex mpi)
                -DUSE_FAKELOWP=OFF
@@ -181,6 +181,8 @@ src_configure() {
 src_install() {
        cmake_src_install
 
+       use cuda && dolib.so "${BUILD_DIR}"/lib/libnvfuser_codegen.so
+
        insinto "/var/lib/${PN}"
        doins "${BUILD_DIR}"/CMakeCache.txt