gentoo-overlay/dev-libs/cudnn/cudnn-8.0.5.39.ebuild

# Copyright 1999-2020 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2

EAPI=7

DESCRIPTION="NVIDIA Accelerated Deep Learning on GPU library"
HOMEPAGE="https://developer.nvidia.com/cuDNN"

SRC_URI="cuda11-1? ( cudnn-11.1-linux-x64-v${PV}.tgz )
	cuda11-0? ( cudnn-11.0-linux-x64-v${PV}.tgz )
	cuda10-2? ( cudnn-10.2-linux-x64-v${PV}.tgz )
	cuda10-1? ( cudnn-10.1-linux-x64-v${PV}.tgz )"

SLOT="0/8"
KEYWORDS="~amd64 ~amd64-linux"
RESTRICT="fetch"
LICENSE="NVIDIA-cuDNN"
QA_PREBUILT="*"

IUSE="cuda10-1 cuda10-2 cuda11-0 +cuda11-1"
REQUIRED_USE="^^ ( cuda10-2 cuda10-1 cuda11-0 cuda11-1 )"

S="${WORKDIR}"

DEPEND="cuda11-1? ( =dev-util/nvidia-cuda-toolkit-11.1* )
	cuda11-0? ( =dev-util/nvidia-cuda-toolkit-11.0* )
	cuda10-2? ( =dev-util/nvidia-cuda-toolkit-10.2* )
	cuda10-1? ( =dev-util/nvidia-cuda-toolkit-10.1* )"
RDEPEND="${DEPEND}"

src_install() {
	insinto /opt/cuda
	doins cuda/NVIDIA_SLA_cuDNN_Support.txt

	insinto /opt/cuda/targets/x86_64-linux/include
	doins -r cuda/include/*

	insinto /opt/cuda/targets/x86_64-linux/lib
	doins -r cuda/lib*/*
}