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

[Alfredo Tupone]

commit:     71578f583267ca3847dab44526e40eb0ca171dec
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Fri Aug 11 08:39:37 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Fri Aug 11 08:40:05 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=71578f58

sci-libs/caffe2: build nvfuser for cuda users

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

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

diff --git a/sci-libs/caffe2/caffe2-2.0.1-r3.ebuild 
b/sci-libs/caffe2/caffe2-2.0.1-r4.ebuild
similarity index 98%
rename from sci-libs/caffe2/caffe2-2.0.1-r3.ebuild
rename to sci-libs/caffe2/caffe2-2.0.1-r4.ebuild
index 29087a44643f..67ab76e8ed76 100644
--- a/sci-libs/caffe2/caffe2-2.0.1-r3.ebuild
+++ b/sci-libs/caffe2/caffe2-2.0.1-r4.ebuild
@@ -136,7 +136,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
@@ -196,6 +196,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