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

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

[Alfredo Tupone]

commit:     c20b2f88a074632d3741ae5b3f498483e84fe388
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Sun May 14 17:07:11 2023 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Sun May 14 17:07:52 2023 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=c20b2f88

sci-libs/tensorpipe: enable cuda

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

 sci-libs/tensorpipe/metadata.xml                           |  3 +++
 ...e-2022.05.13.ebuild => tensorpipe-2022.05.13-r1.ebuild} | 14 ++++++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/sci-libs/tensorpipe/metadata.xml b/sci-libs/tensorpipe/metadata.xml
index a676d5dbb87a..0771cea93d5e 100644
--- a/sci-libs/tensorpipe/metadata.xml
+++ b/sci-libs/tensorpipe/metadata.xml
@@ -5,6 +5,9 @@
 		<email>tupone@gentoo.org</email>
 		<name>Tupone Alfredo</name>
 	</maintainer>
+	<use>
+		<flag name="cuda">Add support for CUDA processing</flag>
+	</use>
 	<upstream>
 		<remote-id type="github">pytorch/tensorpipe</remote-id>
 	</upstream>

diff --git a/sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild b/sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
similarity index 73%
rename from sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild
rename to sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
index 23422f9cc125..734a2050cf07 100644
--- a/sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild
+++ b/sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
@@ -1,4 +1,4 @@
-# Copyright 2022 Gentoo Authors
+# Copyright 2022-2023 Gentoo Authors
 # Distributed under the terms of the GNU General Public License v2
 
 EAPI=8
@@ -15,16 +15,26 @@ SRC_URI="https://github.com/pytorch/${PN}/archive/${CommitId}.tar.gz
 LICENSE="BSD"
 SLOT="0"
 KEYWORDS="~amd64"
+IUSE="cuda"
 RESTRICT=test
 
 RDEPEND="
 	dev-libs/libuv
+	cuda? (
+		dev-util/nvidia-cuda-toolkit:=
+	)
 "
 DEPEND="${RDEPEND}
 	dev-libs/libnop
 "
-BDEPEND=""
 
 S="${WORKDIR}"/${PN}-${CommitId}
 
 PATCHES=( "${FILESDIR}"/${P}-gentoo.patch )
+
+src_configure() {
+	local mycmakeargs=(
+		-DTP_USE_CUDA=$(usex cuda)
+	)
+	cmake_src_configure
+}