[gentoo-dev] New eclass cuda.eclass

[Justin]

Hi,

I would like to introduce a new eclass for packages using the nvidia
cuda compiler suite. Currently the eclass simply sanitize the NVCCFLAGS.
May be extended in the future.

Two problems come up with using nvcc:

* Each version only supports a limited number of gcc versions. Therefore
we need to pass the path to a supported gcc bindir

* nvcc calls CXX but doesn't pass CXXFLAGS on.


justin

# Copyright 1999-20012 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Header: $

inherit toolchain-funcs versionator

# @ECLASS: cuda.eclass
# @MAINTAINER:
# Justin Lecher <j...@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.
#
# Common usage:
#
# inherit cuda

# @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}

[[ "${CUDA_VERBOSE}" == true ]] && NVCCFLAGS+=" -v"

# @ECLASS-FUNCTION: cuda_gccdir
# @DESCRIPTION:
# Helper for determination of the latest gcc bindir supported by
# then current nvidia cuda toolkit.
#
# Calling plain it returns simply the path, but you probably want to add \"-f\""
# to get the full flag to add to nvcc call.
#
# Example:
#
# cuda_gccdir -f
# -> --compiler-bindir="/usr/x86_64-pc-linux-gnu/gcc-bin/4.6.3"
cuda_gccdir() {
        local _gcc_bindir _ver _args="" _flag _ret

        # Currently we only support the gnu compiler suite
        if [[ $(tc-getCXX) != *g++* ]]; 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 [[ ! $(type -P cuda-config) ]]; 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 $(cuda-config -s))"
                if [[ ! -n ${_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
}

# @ECLASS-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() {
        # 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}\""

        export NVCCFLAGS
}

# @ECLASS-FUNCTION: cuda_pkg_setup
# @DESCRIPTION:
# Sanitise and export NVCCFLAGS by default in pkg_setup
cuda_pkg_setup() {
        cuda_sanitize
}

EXPORT_FUNCTIONS pkg_setup
case "${EAPI:-0}" in
   0|1|2|3|4|5) ;;
   *) die "EAPI=${EAPI} is not supported" ;;
esac

signature.asc
Description: OpenPGP digital signature