Adding externals support for CUDA. (#18011)
Co-authored-by: Massimiliano Culpo <massimiliano.culpo@gmail.com>
This commit is contained in:
parent
193c3535e1
commit
b4ff584bc0
1 changed files with 10 additions and 0 deletions
|
@ -7,6 +7,7 @@
|
|||
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
|
||||
|
|
Loading…
Reference in a new issue