From 4f2d7949f03e1c198bc888f2d05f421d35c57e21 Mon Sep 17 00:00:00 2001
From: V3n3RiX <venerix@redcorelinux.org>
Date: Mon, 9 Oct 2017 18:53:29 +0100
Subject: reinit the tree, so we can have metadata

---
 eclass/cuda.eclass | 145 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 145 insertions(+)
 create mode 100644 eclass/cuda.eclass

(limited to 'eclass/cuda.eclass')

diff --git a/eclass/cuda.eclass b/eclass/cuda.eclass
new file mode 100644
index 000000000000..9ae7a28e1d2b
--- /dev/null
+++ b/eclass/cuda.eclass
@@ -0,0 +1,145 @@
+# Copyright 1999-2016 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+
+inherit flag-o-matic toolchain-funcs versionator
+
+# @ECLASS: cuda.eclass
+# @MAINTAINER:
+# Justin Lecher <jlec@gentoo.org>
+# @BLURB: Common functions for cuda packages
+# @DESCRIPTION:
+# This eclass contains functions to be used with cuda package. Currently it is
+# setting and/or sanitizing NVCCFLAGS, the compiler flags for nvcc. This is
+# automatically done and exported in src_prepare() or manually by calling
+# cuda_sanatize.
+# @EXAMPLE:
+# inherit cuda
+
+if [[ -z ${_CUDA_ECLASS} ]]; then
+
+# @ECLASS-VARIABLE: NVCCFLAGS
+# @DESCRIPTION:
+# nvcc compiler flags (see nvcc --help), which should be used like
+# CFLAGS for c compiler
+: ${NVCCFLAGS:=-O2}
+
+# @ECLASS-VARIABLE: CUDA_VERBOSE
+# @DESCRIPTION:
+# Being verbose during compilation to see underlying commands
+: ${CUDA_VERBOSE:=true}
+
+# @FUNCTION: cuda_gccdir
+# @USAGE: [-f]
+# @RETURN: gcc bindir compatible with current cuda, optionally (-f) prefixed with "--compiler-bindir "
+# @DESCRIPTION:
+# Helper for determination of the latest gcc bindir supported by
+# then current nvidia cuda toolkit.
+#
+# Example:
+# @CODE
+# cuda_gccdir -f
+# -> --compiler-bindir "/usr/x86_64-pc-linux-gnu/gcc-bin/4.6.3"
+# @CODE
+cuda_gccdir() {
+	debug-print-function ${FUNCNAME} "$@"
+
+	local gcc_bindir ver args="" flag ret
+
+	# Currently we only support the gnu compiler suite
+	if  ! tc-is-gcc ; then
+		ewarn "Currently we only support the gnu compiler suite"
+		return 2
+	fi
+
+	while [ "$1" ]; do
+		case $1 in
+			-f)
+				flag="--compiler-bindir "
+				;;
+			*)
+				;;
+		esac
+		shift
+	done
+
+	if ! args=$(cuda-config -s); then
+		eerror "Could not execute cuda-config"
+		eerror "Make sure >=dev-util/nvidia-cuda-toolkit-4.2.9-r1 is installed"
+		die "cuda-config not found"
+	else
+		args=$(version_sort ${args})
+		if [[ -z ${args} ]]; then
+			die "Could not determine supported gcc versions from cuda-config"
+		fi
+	fi
+
+	for ver in ${args}; do
+		has_version "=sys-devel/gcc-${ver}*" && \
+		 gcc_bindir="$(ls -d ${EPREFIX}/usr/*pc-linux-gnu/gcc-bin/${ver}* | tail -n 1)"
+	done
+
+	if [[ -n ${gcc_bindir} ]]; then
+		if [[ -n ${flag} ]]; then
+			ret="${flag}\"${gcc_bindir}\""
+		else
+			ret="${gcc_bindir}"
+		fi
+		echo ${ret}
+		return 0
+	else
+		eerror "Only gcc version(s) ${args} are supported,"
+		eerror "of which none is installed"
+		die "Only gcc version(s) ${args} are supported"
+		return 1
+	fi
+}
+
+# @FUNCTION: cuda_sanitize
+# @DESCRIPTION:
+# Correct NVCCFLAGS by adding the necessary reference to gcc bindir and
+# passing CXXFLAGS to underlying compiler without disturbing nvcc.
+cuda_sanitize() {
+	debug-print-function ${FUNCNAME} "$@"
+
+	local rawldflags=$(raw-ldflags)
+	# Be verbose if wanted
+	[[ "${CUDA_VERBOSE}" == true ]] && NVCCFLAGS+=" -v"
+
+	# Tell nvcc where to find a compatible compiler
+	NVCCFLAGS+=" $(cuda_gccdir -f)"
+
+	# Tell nvcc which flags should be used for underlying C compiler
+	NVCCFLAGS+=" --compiler-options \"${CXXFLAGS}\" --linker-options \"${rawldflags// /,}\""
+
+	debug-print "Using ${NVCCFLAGS} for cuda"
+	export NVCCFLAGS
+}
+
+# @FUNCTION: cuda_pkg_setup
+# @DESCRIPTION:
+# Call cuda_src_prepare for EAPIs not supporting src_prepare
+cuda_pkg_setup() {
+	debug-print-function ${FUNCNAME} "$@"
+
+	cuda_src_prepare
+}
+
+# @FUNCTION: cuda_src_prepare
+# @DESCRIPTION:
+# Sanitise and export NVCCFLAGS by default
+cuda_src_prepare() {
+	debug-print-function ${FUNCNAME} "$@"
+
+	cuda_sanitize
+}
+
+case "${EAPI:-0}" in
+	0|1)
+		EXPORT_FUNCTIONS pkg_setup ;;
+	2|3|4|5|6)
+		EXPORT_FUNCTIONS src_prepare ;;
+	*) die "EAPI=${EAPI} is not supported" ;;
+esac
+
+_CUDA_ECLASS=1
+fi
-- 
cgit v1.2.3