diff options
Diffstat (limited to 'sci-libs/tensorpipe')
| -rw-r--r-- | sci-libs/tensorpipe/Manifest | 4 | ||||
| -rw-r--r-- | sci-libs/tensorpipe/metadata.xml | 3 | ||||
| -rw-r--r-- | sci-libs/tensorpipe/tensorpipe-2022.05.13-r1.ebuild (renamed from sci-libs/tensorpipe/tensorpipe-2022.05.13.ebuild) | 14 |
3 files changed, 17 insertions, 4 deletions
diff --git a/sci-libs/tensorpipe/Manifest b/sci-libs/tensorpipe/Manifest index c5bb13a8fdf1..694912c39de3 100644 --- a/sci-libs/tensorpipe/Manifest +++ b/sci-libs/tensorpipe/Manifest @@ -1,4 +1,4 @@ AUX tensorpipe-2022.05.13-gentoo.patch 522 BLAKE2B b5efc8193549a84569125794e65a931343f26996f487158f39dd6c9d1689ecd1390bae1616e1f498461a3b4b0a17478a8a14c1cbd6b76c82b7dc2a4452e72d7e SHA512 3c8cd91daf9cf1c4249c5e5f7154f9d4495d33ab011802f83a96d056723ebfb86406f4bc9c8bacc8af2a70336d706de9d88d538d196348d2d6ea4e56886b5162 DIST tensorpipe-2022.05.13.tar.gz 259678 BLAKE2B 09231216b7ea1a5f4559bfd38d82337cdcde160c34ceb160c728b85ccf94566b812f0c38b6d3f6bc74b6f08b49b749939970265df343e3ca0214803f9d88e58c SHA512 cf0334f81affb2d844bc8b63c533a749753e36ee096f841641716a3bf044b17580262a2e9056d8d1351228e323c4f75401a2a120a5de397e80ec21a33fe56d2b -EBUILD tensorpipe-2022.05.13.ebuild 569 BLAKE2B 70d536748302b8a0fa0fcbbc96b45901ab242bf0eaa3f7d2ab92da0280fc7a252673dc62ae426c3844bf52817e9456087d9eda270659784d96b7e207d7fb9449 SHA512 517023c737aed9ab89783f28c88b4cb68c5bd4bfb90dba56fe99edf4d585addfafd47e4efb352e7a9a6ffe0bd322a17a6396feaf0998bb7d435d3d47160e726e -MISC metadata.xml 331 BLAKE2B 6a5a854e8e6c987dc1b105a3c905d8113b0613e63f479e700f85a01b96e4b939de8706bc876c1bf44bd61f40b9b3b18eb87f3899c1f3697ddfb5ca6bed2a7a61 SHA512 8843f661956cdbfbda49aebfe4586c5eaafa7e7b82f134d5881e46ac6ebf69b20167f532a372280be17ae29016aca82e2930762b2bdbb7616918b7200a3a2a49 +EBUILD tensorpipe-2022.05.13-r1.ebuild 715 BLAKE2B acf19eb22134d1a77d7b421702a98e907104747c87afb16253680c72dc000ae10ef2e356678384e71522032e1ad52f8fc0e6f5f26332c0d30b459c87a2e99bc7 SHA512 b84ba32b1f4abe94db73418625184f0a843170a6604dcaa9a855a3905d09efeb82cdfbc461248cd9b4cfbf740b5aa3001b504d3adacf2ae7b8404a703d78830c +MISC metadata.xml 405 BLAKE2B 8204155408fdc2a096eb6bfc28958c0b355d15f3c0afe30143b6919a0db1762d77c9e7048de5dcb210514fd56c2cad387e1bed365b66d51bf77917813fd27496 SHA512 2810baff83febd987e2c7d66a57e491ad7e13be65c0eb965955fce00b2f62811938c1aa9b9401b1764b710ca573c57a42fe328de6de7060aa0e0ab8d481fe1aa 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 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 +} |