diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/cuda/package.py | 10 |
1 files changed, 10 insertions, 0 deletions
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 |