From b4ff584bc03b054ad77741f8c5db57cb023f62e1 Mon Sep 17 00:00:00 2001 From: Robert Blake Date: Wed, 12 Aug 2020 10:30:57 -0700 Subject: Adding externals support for CUDA. (#18011) Co-authored-by: Massimiliano Culpo --- var/spack/repos/builtin/packages/cuda/package.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/var/spack/repos/builtin/packages/cuda/package.py b/var/spack/repos/builtin/packages/cuda/package.py index adf7bf8900..785f7c3c0c 100644 --- a/var/spack/repos/builtin/packages/cuda/package.py +++ b/var/spack/repos/builtin/packages/cuda/package.py @@ -7,6 +7,7 @@ from spack import * from glob import glob from llnl.util.filesystem import LibraryList import os +import re import platform import llnl.util.tty as tty @@ -60,6 +61,8 @@ class Cuda(Package): homepage = "https://developer.nvidia.com/cuda-zone" + executables = ['^nvcc$'] + for ver, packages in _versions.items(): key = "{0}-{1}".format(platform.system(), platform.machine()) pkg = packages.get(key) @@ -79,6 +82,13 @@ class Cuda(Package): depends_on('libxml2', when='@10.1.243:') + @classmethod + def determine_version(cls, exe): + output = Executable(exe)('--version', output=str, error=str) + match = re.search(r'Cuda compilation tools, release .*?, V(\S+)', + output) + return match.group(1) if match else None + def setup_build_environment(self, env): if self.spec.satisfies('@10.1.243:'): libxml2_home = self.spec['libxml2'].prefix -- cgit v1.2.3-70-g09d2