commit:     04f1c73d1190c41edc24667ebd0c2505c62a6706
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Mon Feb 27 07:13:55 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Mon Feb 27 07:23:15 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=04f1c73d

sci-libs/caffe2: find the right GCC for CUDA

Closes: https://bugs.gentoo.org/888039
Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-libs/caffe2/caffe2-1.13.1-r1.ebuild | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/sci-libs/caffe2/caffe2-1.13.1-r1.ebuild 
b/sci-libs/caffe2/caffe2-1.13.1-r1.ebuild
index 352a593d3559..4e0dfcb9b980 100644
--- a/sci-libs/caffe2/caffe2-1.13.1-r1.ebuild
+++ b/sci-libs/caffe2/caffe2-1.13.1-r1.ebuild
@@ -4,7 +4,7 @@
 EAPI=8
 
 PYTHON_COMPAT=( python3_{9..11} )
-inherit python-single-r1 cmake flag-o-matic
+inherit python-single-r1 cmake cuda flag-o-matic
 
 MYPN=pytorch
 MYP=${MYPN}-${PV}
@@ -148,7 +148,13 @@ src_configure() {
                -DLIBSHM_INSTALL_LIB_SUBDIR="${EPREFIX}"/usr/$(get_libdir)
        )
 
-       use cuda && addpredict "/dev/nvidiactl" # bug 867706
+       if use cuda; then
+               addpredict "/dev/nvidiactl" # bug 867706
+
+               mycmakeargs+=(
+                       -DCMAKE_CUDA_FLAGS="$(cuda_gccdir -f | tr -d \")"
+               )
+       fi
        cmake_src_configure
 }
 

Reply via email to