sci-ml/cudnn-frontend: add 1.16.1

Signed-off-by: Paul Zander <negril.nx+gentoo@gmail.com>
Part-of: https://github.com/gentoo/gentoo/pull/45104
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
This commit is contained in:
Paul Zander 2025-12-19 15:04:38 +01:00 committed by Alfredo Tupone
parent 6c6c2ae391
commit 8596d347fa
No known key found for this signature in database
GPG Key ID: FBE5925C5B02CE58
2 changed files with 108 additions and 0 deletions

View File

@ -2,3 +2,4 @@ DIST cudnn-frontend-1.10.0.tar.gz 662344 BLAKE2B f151a57afdf41c03829b67d9ecda0e7
DIST cudnn-frontend-1.11.0.tar.gz 653191 BLAKE2B 2f784424bb4c850683b8ecf0b24e03c761227f839d63ad92eba38b5d9a530800a32e7311f27c0bc9e31a0bfc26324acf4d194d143f69f1b6094a3958d2874ce0 SHA512 2d50fbedc1d2f488275aedce84893447a025d4c00b9e8609c4004b2eb0525480a348835d0e8b2784499d80d0c63d75bb1430741cb06c3652da8dd72b822489fa
DIST cudnn-frontend-1.12.1.tar.gz 1683639 BLAKE2B 1d00e0caea27fc63883b867fa192f1346c67318f4781b0255e0203753a7ead80746291736e6972c58aa8a1870ccac38ea4dbfe551e5da97596a857d53bc826ea SHA512 02dba5b4480099d1e14abb2fc16d64466955988932bbfdabfe58104576582c644cacec07674baf23e51bb3d8311e5b7080b1e3b3c140d9f463767c2f117e750e
DIST cudnn-frontend-1.15.0.tar.gz 1945364 BLAKE2B 0881cef7635d32145d396fd7476c781acf0d6913ff0704552adbb087276cc36d32d53f56a8d1975f7c8b04bf6d5df5e3bbcfbf0ed7dd8d21e6e082a5e0b1bceb SHA512 75943654652bbde8dc90cf52b58078a8c8250940827cebfe96c36750bc41047d016a94f1201f9e4069cdac56e0804c2f5a534ca86d4dc0f0ba4e65a97792a7a7
DIST cudnn-frontend-1.16.1.tar.gz 1981715 BLAKE2B b8112bc538dba7be1596c5e13e37b031f2e158420d6c80e39e3fdb3b8b87600158eda4f30ea6a20df6a41fa04976200916810cf0c6025b2c32357a177a020a30 SHA512 f7e0d234bda6a1d284a89974ccd1e0e13e87af774372a41d4fb3af43b01a2a73eddc2deb43d2726f536238366124c64cbd48cbb98576550a0fd635815456b008

View File

@ -0,0 +1,107 @@
# Copyright 1999-2025 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
inherit cuda cmake edo
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"
IUSE="samples test"
RESTRICT="!test? ( test )"
RDEPEND="
>=dev-libs/cudnn-9.0.0:=
"
DEPEND="${RDEPEND}
dev-cpp/nlohmann_json
test? (
>=dev-cpp/catch-3
>=dev-libs/cudnn-9.15.0
)
"
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
addwrite "/proc/self/task"
# List all tests
# "${BUILD_DIR}/bin/tests" --list-tests --verbosity quiet
local catchargs=()
local CATCH_SKIP_TESTS=()
if [[ -v CUDAARCHS && "${CUDAARCHS}" != 89 ]]; then
CATCH_SKIP_TESTS+=(
# doesn't work on 86 52
'Graph key'
'Graph key dynamic shape'
# doesn't work on 86
'sdpa backward graph serialization'
)
fi
[[ -v CATCH_SKIP_TESTS ]] && catchargs+=( "${CATCH_SKIP_TESTS[@]/#/\~}" )
edo "${BUILD_DIR}/bin/tests" -s "${catchargs[@]}"
if use samples; then
edo "${BUILD_DIR}/bin/samples" -s
edo "${BUILD_DIR}/bin/legacy_samples" -s
fi
cmake_src_test
}
src_install() {
cmake_src_install
if use test; then
rm -R "${ED}/usr/bin/tests" || die
fi
}