gentoo-ebuilds/sci-ml/tensorpipe/tensorpipe-2022.05.13-r1.ebuild
Alfredo Tupone 054d60f7ca
sci-ml/*: mv sci-libs/tensorpipe to sci-ml/
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
2025-03-15 17:21:34 +01:00

44 lines
779 B
Bash

# Copyright 2022-2025 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"
S="${WORKDIR}"/${PN}-${CommitId}
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
"
PATCHES=(
"${FILESDIR}"/${P}-gentoo.patch
"${FILESDIR}"/${P}-musl.patch
"${FILESDIR}"/${P}-gcc15.patch
)
src_configure() {
local mycmakeargs=(
-DTP_USE_CUDA=$(usex cuda)
)
cmake_src_configure
}