From c20b2f88a074632d3741ae5b3f498483e84fe388 Mon Sep 17 00:00:00 2001
From: Alfredo Tupone <tupone@gentoo.org>
Date: Sun, 14 May 2023 19:07:11 +0200
Subject: sci-libs/tensorpipe: enable cuda

Closes: https://bugs.gentoo.org/906204
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
---
 sci-libs/tensorpipe/metadata.xml                   |  3 ++
 .../tensorpipe/tensorpipe-2022.05.13-r1.ebuild     | 40 ++++++++++++++++++++++
 sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild   | 30 ----------------
 3 files changed, 43 insertions(+), 30 deletions(-)
 create mode 100644 sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
 delete mode 100644 sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild

(limited to 'sci-libs/tensorpipe')

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-r1.ebuild b/sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
new file mode 100644
index 000000000000..734a2050cf07
--- /dev/null
+++ b/sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
@@ -0,0 +1,40 @@
+# Copyright 2022-2023 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+
+inherit cmake
+
+CommitId=bb1473a4b38b18268e8693044afdb8635bc8351b
+
+DESCRIPTION="provides a tensor-aware channel"
+HOMEPAGE="https://github.com/pytorch/tensorpipe/"
+SRC_URI="https://github.com/pytorch/${PN}/archive/${CommitId}.tar.gz
+	-> ${P}.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
+"
+
+S="${WORKDIR}"/${PN}-${CommitId}
+
+PATCHES=( "${FILESDIR}"/${P}-gentoo.patch )
+
+src_configure() {
+	local mycmakeargs=(
+		-DTP_USE_CUDA=$(usex cuda)
+	)
+	cmake_src_configure
+}
diff --git a/sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild b/sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild
deleted file mode 100644
index 23422f9cc125..000000000000
--- a/sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild
+++ /dev/null
@@ -1,30 +0,0 @@
-# Copyright 2022 Gentoo Authors
-# Distributed under the terms of the GNU General Public License v2
-
-EAPI=8
-
-inherit cmake
-
-CommitId=bb1473a4b38b18268e8693044afdb8635bc8351b
-
-DESCRIPTION="provides a tensor-aware channel"
-HOMEPAGE="https://github.com/pytorch/tensorpipe/"
-SRC_URI="https://github.com/pytorch/${PN}/archive/${CommitId}.tar.gz
-	-> ${P}.tar.gz"
-
-LICENSE="BSD"
-SLOT="0"
-KEYWORDS="~amd64"
-RESTRICT=test
-
-RDEPEND="
-	dev-libs/libuv
-"
-DEPEND="${RDEPEND}
-	dev-libs/libnop
-"
-BDEPEND=""
-
-S="${WORKDIR}"/${PN}-${CommitId}
-
-PATCHES=( "${FILESDIR}"/${P}-gentoo.patch )
-- 
cgit v1.2.3-65-gdbad