164 lines
4.1 KiB
Bash
164 lines
4.1 KiB
Bash
# Copyright 1999-2020 Gentoo Authors
|
|
# Distributed under the terms of the GNU General Public License v2
|
|
|
|
EAPI=7
|
|
|
|
inherit check-reqs cuda toolchain-funcs unpacker
|
|
|
|
MYD=$(ver_cut 1-2 ${PV})
|
|
DRIVER_PV="418.87.00"
|
|
|
|
DESCRIPTION="NVIDIA CUDA Toolkit (compiler and friends)"
|
|
HOMEPAGE="https://developer.nvidia.com/cuda-zone"
|
|
SRC_URI="https://developer.download.nvidia.com/compute/cuda/${MYD}/Prod/local_installers/cuda_${PV}_${DRIVER_PV}_linux.run"
|
|
|
|
LICENSE="NVIDIA-CUDA"
|
|
SLOT="0/${PV}"
|
|
KEYWORDS="-* ~amd64 ~amd64-linux"
|
|
IUSE="debugger doc eclipse profiler"
|
|
RESTRICT="bindist mirror"
|
|
|
|
BDEPEND=""
|
|
RDEPEND="
|
|
<sys-devel/gcc-9[cxx]
|
|
!prefix? ( >=x11-drivers/nvidia-drivers-${DRIVER_PV}[X,uvm] )
|
|
debugger? (
|
|
dev-libs/openssl-compat:1.0.0
|
|
sys-libs/libtermcap-compat
|
|
sys-libs/ncurses-compat:5[tinfo]
|
|
)
|
|
eclipse? (
|
|
dev-libs/openssl-compat:1.0.0
|
|
>=virtual/jre-1.6
|
|
)
|
|
profiler? (
|
|
dev-libs/openssl-compat:1.0.0
|
|
>=virtual/jre-1.6
|
|
)"
|
|
|
|
S="${WORKDIR}"
|
|
|
|
QA_PREBUILT="opt/cuda/*"
|
|
|
|
CHECKREQS_DISK_BUILD="6800M"
|
|
|
|
pkg_setup() {
|
|
# We don't like to run cuda_pkg_setup as it depends on us
|
|
check-reqs_pkg_setup
|
|
}
|
|
|
|
src_prepare() {
|
|
local cuda_supported_gcc
|
|
|
|
cuda_supported_gcc="4.7 4.8 4.9 5.3 5.4 6.3 6.4 7.2 7.3 8.2 8.3"
|
|
|
|
sed \
|
|
-e "s:CUDA_SUPPORTED_GCC:${cuda_supported_gcc}:g" \
|
|
"${FILESDIR}"/cuda-config.in > "${T}"/cuda-config || die
|
|
|
|
default
|
|
}
|
|
|
|
src_install() {
|
|
local i remove=( doc )
|
|
local cudadir=/opt/cuda
|
|
local ecudadir="${EPREFIX}${cudadir}"
|
|
|
|
cd builds/cuda-toolkit
|
|
if use doc; then
|
|
DOCS+=( doc/pdf/. )
|
|
HTML_DOCS+=( doc/html/. )
|
|
fi
|
|
einstalldocs
|
|
|
|
mv doc/man/man3/{,cuda-}deprecated.3 || die
|
|
doman doc/man/man*/*
|
|
|
|
use debugger || remove+=( bin/cuda-gdb bin/cuda-gdbserver extras/Debugger share/gdb extras/cuda-gdb-${PV}.src.tar.gz )
|
|
|
|
if use profiler; then
|
|
# hack found in install-linux.pl
|
|
for i in nvvp nsight; do
|
|
cat > bin/${i} <<- EOF || die
|
|
#!/usr/bin/env sh
|
|
LD_LIBRARY_PATH=\${LD_LIBRARY_PATH}:${ecudadir}/lib:${ecudadir}/lib64 \
|
|
UBUNTU_MENUPROXY=0 LIBOVERLAY_SCROLLBAR=0 \
|
|
${ecudadir}/lib${i}/${i} -vm ${EPREFIX}/usr/bin/java
|
|
EOF
|
|
chmod a+x bin/${i} || die
|
|
done
|
|
else
|
|
use eclipse || remove+=( libnvvp libnsight nsightee_plugins nsight-compute-2019.4.0 nsight-systems-2019.3.7.5 bin/nsight bin/nsight_ee_plugins_manage.sh bin/nvvp bin/computeprof )
|
|
remove+=( extras/CUPTI bin/nvprof )
|
|
fi
|
|
|
|
for i in "${remove[@]}"; do
|
|
ebegin "Cleaning ${i}..."
|
|
rm -r "${i}" || die
|
|
eend
|
|
done
|
|
|
|
dodir ${cudadir}
|
|
into ${cudadir}
|
|
|
|
# Install binaries separately to make sure the X permission is set
|
|
local bindirs=( bin nvvm/bin extras/demo_suite $(usex profiler "libnsight/nsight") )
|
|
for i in $(find "${bindirs[@]}" -maxdepth 1 -type f); do
|
|
exeinto ${cudadir}/${i%/*}
|
|
doexe ${i}
|
|
rm ${i} || die
|
|
done
|
|
exeinto ${cudadir}/bin
|
|
doexe "${T}"/cuda-config
|
|
|
|
# Install the rest
|
|
insinto ${cudadir}
|
|
doins -r *
|
|
|
|
cat > "${T}"/99cuda <<- EOF || die
|
|
PATH=${ecudadir}/bin$(usex profiler ":${ecudadir}/libnvvp" "")
|
|
ROOTPATH=${ecudadir}/bin
|
|
LDPATH=${ecudadir}/lib64:${ecudadir}/lib:${ecudadir}/nvvm/lib64
|
|
EOF
|
|
doenvd "${T}"/99cuda
|
|
|
|
#Cuda prepackages libraries, don't revdep-build on them
|
|
echo "SEARCH_DIRS_MASK=\"${ecudadir}\"" > "${T}/80${PN}" || die
|
|
insinto "/etc/revdep-rebuild"
|
|
doins "${T}/80${PN}"
|
|
}
|
|
|
|
pkg_postinst_check() {
|
|
local a b
|
|
a="$(${EROOT}/opt/cuda/bin/cuda-config -s)"
|
|
b="0.0"
|
|
for v in $a; do
|
|
if ver_test "${v}" -gt "${b}"; then
|
|
b="${v}"
|
|
fi
|
|
done
|
|
|
|
# if gcc and if not gcc-version is at least greatest supported
|
|
if tc-is-gcc && \
|
|
ver_test $(gcc-version) -gt ${b}; then
|
|
ewarn ""
|
|
ewarn "gcc > ${b} will not work with CUDA"
|
|
ewarn "Make sure you set an earlier version of gcc with gcc-config"
|
|
ewarn "or append --compiler-bindir= pointing to a gcc bindir like"
|
|
ewarn "--compiler-bindir=${EPREFIX}/usr/*pc-linux-gnu/gcc-bin/gcc${b}"
|
|
ewarn "to the nvcc compiler flags"
|
|
ewarn ""
|
|
fi
|
|
}
|
|
|
|
pkg_postinst() {
|
|
if [[ ${MERGE_TYPE} != binary ]]; then
|
|
pkg_postinst_check
|
|
fi
|
|
|
|
if use prefix; then
|
|
ewarn "Gentoo Prefix does not manage kernel modules. You need to make certain"
|
|
ewarn "the function counterpart to >=x11-drivers/nvidia-drivers-${DRIVER_PV}[X,uvm]"
|
|
ewarn "is available from the host"
|
|
fi
|
|
}
|