mirror of
https://anongit.gentoo.org/git/repo/gentoo.git
synced 2025-12-16 23:19:31 +00:00
77 lines
1.5 KiB
Bash
77 lines
1.5 KiB
Bash
# Copyright 1999-2025 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=8
|
|
|
|
inherit cuda cmake
|
|
|
|
DESCRIPTION="A c++ wrapper for the cudnn backend API"
|
|
HOMEPAGE="https://github.com/NVIDIA/cudnn-frontend"
|
|
|
|
SRC_URI="https://github.com/NVIDIA/cudnn-frontend/archive/refs/tags/v${PV}.tar.gz -> ${P}.tar.gz"
|
|
|
|
LICENSE="MIT"
|
|
SLOT="0/$(ver_cut 1-2)"
|
|
KEYWORDS="~amd64 ~amd64-linux"
|
|
IUSE="samples test"
|
|
RESTRICT="!test? ( test )"
|
|
|
|
RDEPEND="
|
|
dev-libs/cudnn
|
|
"
|
|
DEPEND="${RDEPEND}
|
|
dev-cpp/nlohmann_json
|
|
test? (
|
|
>dev-cpp/catch-3
|
|
)
|
|
"
|
|
|
|
PATCHES=(
|
|
"${FILESDIR}/${PN}-1.11.0-fix.patch"
|
|
)
|
|
|
|
src_prepare() {
|
|
cmake_src_prepare
|
|
|
|
sed -e 's#"cudnn_frontend/thirdparty/nlohmann/json.hpp"#<nlohmann/json.hpp>#' \
|
|
-i include/cudnn_frontend_utils.h || die
|
|
|
|
rm -r include/cudnn_frontend/thirdparty || die
|
|
}
|
|
|
|
src_configure() {
|
|
local narch
|
|
if use amd64; then
|
|
narch="x86_64"
|
|
elif use arm64; then
|
|
narch="sbsa"
|
|
fi
|
|
|
|
local mycmakeargs=(
|
|
-DCUDNN_FRONTEND_BUILD_PYTHON_BINDINGS="no"
|
|
-DCUDNN_FRONTEND_BUILD_SAMPLES="$(usex test "$(usex samples)")"
|
|
-DCUDNN_FRONTEND_BUILD_TESTS="$(usex test)"
|
|
-DCUDNN_FRONTEND_SKIP_JSON_LIB="no"
|
|
)
|
|
|
|
if use samples || use test; then
|
|
# allow slotted install
|
|
: "${CUDNN_PATH:=${ESYSROOT}/opt/cuda}"
|
|
export CUDNN_PATH
|
|
fi
|
|
|
|
cmake_src_configure
|
|
}
|
|
|
|
src_test() {
|
|
cuda_add_sandbox -w
|
|
|
|
"${BUILD_DIR}/bin/tests" || die
|
|
|
|
if use samples; then
|
|
"${BUILD_DIR}/bin/samples" || die
|
|
"${BUILD_DIR}/bin/legacy_samples" || die
|
|
fi
|
|
|
|
cmake_src_test
|
|
}
|