diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/spack/spack/build_systems/cuda.py | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/lib/spack/spack/build_systems/cuda.py b/lib/spack/spack/build_systems/cuda.py new file mode 100644 index 0000000000..874b30f679 --- /dev/null +++ b/lib/spack/spack/build_systems/cuda.py @@ -0,0 +1,88 @@ +############################################################################## +# Copyright (c) 2013-2017, Lawrence Livermore National Security, LLC. +# Produced at the Lawrence Livermore National Laboratory. +# +# This file is part of Spack. +# Created by Todd Gamblin, tgamblin@llnl.gov, All rights reserved. +# LLNL-CODE-647188 +# +# For details, see https://github.com/spack/spack +# Please also see the NOTICE and LICENSE files for our notice and the LGPL. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License (as +# published by the Free Software Foundation) version 2.1, February 1999. +# +# This program is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the IMPLIED WARRANTY OF +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the terms and +# conditions of the GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +############################################################################## +from spack.package import Package +from spack.directives import depends_on, variant, conflicts +import platform + + +class CudaPackage(Package): + """Auxiliary class which contains CUDA variant, dependencies and conflicts + and is meant to unify and facilitate its usage. + """ + + # FIXME: keep cuda and cuda_arch separate to make usage easier untill + # Spack has depends_on(cuda, when='cuda_arch!=None') or alike + variant('cuda', default=False, + description='Build with CUDA') + # see http://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#gpu-feature-list + # https://developer.nvidia.com/cuda-gpus + variant('cuda_arch', default=None, + description='CUDA architecture', + values=('20', '30', '32', '35', '50', '52', '53', '60', '61', + '62', '70'), + multi=True) + + # see http://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/index.html#nvcc-examples + # and http://llvm.org/docs/CompileCudaWithLLVM.html#compiling-cuda-code + @staticmethod + def cuda_flags(arch_list): + return [('--generate-code arch=compute_{0},code=sm_{0} ' + '--generate-code arch=compute_{0},code=compute_{0}').format(s) + for s in arch_list] + + depends_on("cuda@7:", when='+cuda') + + # CUDA version vs Architecture + depends_on("cuda@8:", when='cuda_arch=60') + depends_on("cuda@8:", when='cuda_arch=61') + depends_on("cuda@8:", when='cuda_arch=62') + depends_on("cuda@9:", when='cuda_arch=70') + + depends_on('cuda@:8.99', when='cuda_arch=20') + + # Compiler conflicts: + # https://gist.github.com/ax3l/9489132 + conflicts('%gcc@5:', when='+cuda ^cuda@:7.5') + conflicts('%gcc@6:', when='+cuda ^cuda@:8.99') + conflicts('%gcc@7:', when='+cuda ^cuda@:9.99') + if (platform.system() != "Darwin"): + conflicts('%clang@:3.4,3.7:', when='+cuda ^cuda@7.5') + conflicts('%clang@:3.7,4:', when='+cuda ^cuda@8:9') + conflicts('%intel@:14,16:', when='+cuda ^cuda@7.5') + conflicts('%intel@:14,17:', when='+cuda ^cuda@8.0.44') + conflicts('%intel@:14,18:', when='+cuda ^cuda@8.0.61:9') + + # Make sure cuda_arch can not be used without +cuda + conflicts('~cuda', when='cuda_arch=20') + conflicts('~cuda', when='cuda_arch=30') + conflicts('~cuda', when='cuda_arch=32') + conflicts('~cuda', when='cuda_arch=35') + conflicts('~cuda', when='cuda_arch=50') + conflicts('~cuda', when='cuda_arch=52') + conflicts('~cuda', when='cuda_arch=53') + conflicts('~cuda', when='cuda_arch=60') + conflicts('~cuda', when='cuda_arch=61') + conflicts('~cuda', when='cuda_arch=62') + conflicts('~cuda', when='cuda_arch=70') |