diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/blaspp/package.py')
| -rw-r--r-- | var/spack/repos/builtin/packages/blaspp/package.py | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/var/spack/repos/builtin/packages/blaspp/package.py b/var/spack/repos/builtin/packages/blaspp/package.py index 5361013681..1013a954a0 100644 --- a/var/spack/repos/builtin/packages/blaspp/package.py +++ b/var/spack/repos/builtin/packages/blaspp/package.py @@ -7,7 +7,7 @@ from spack import * import os -class Blaspp(CMakePackage, CudaPackage): +class Blaspp(CMakePackage, CudaPackage, ROCmPackage): """C++ API for the Basic Linear Algebra Subroutines. Developed by the Innovative Computing Laboratory at the University of Tennessee, Knoxville.""" @@ -18,39 +18,52 @@ class Blaspp(CMakePackage, CudaPackage): maintainers = ['teonnik', 'Sely85', 'G-Ragghianti', 'mgates3'] version('master', branch='master') + version('2021.04.01', sha256='11fc7b7e725086532ada58c0de53f30e480c2a06f1497b8081ea6d8f97e26150') version('2020.10.02', sha256='36e45bb5a8793ba5d7bc7c34fc263f91f92b0946634682937041221a6bf1a150') version('2020.10.01', sha256='1a05dbc46caf797d59a7c189216b876fdb1b2ff3e2eb48f1e6ca4b2756c59153') version('2020.10.00', sha256='ce148cfe397428d507c72d7d9eba5e9d3f55ad4cd842e6e873c670183dcb7795') variant('openmp', default=True, description='Use OpenMP internally.') - variant('cuda', default=False, description='Build with CUDA') variant('shared', default=True, description='Build shared libraries') depends_on('cmake@3.15.0:', type='build') depends_on('blas') depends_on('llvm-openmp', when='%apple-clang +openmp') + depends_on('rocblas', when='+rocm') # only supported with clingo solver: virtual dependency preferences # depends_on('openblas threads=openmp', when='+openmp ^openblas') - # BLASpp tests will fail when using openblas > 0.3.5 without multithreading support + # BLAS++ tests will fail when using openblas > 0.3.5 without multithreading support # locking is only supported in openblas 3.7+ - conflicts('^openblas@0.3.6 threads=none', msg='BLASpp requires a threadsafe openblas') - conflicts('^openblas@0.3.7: ~locking', msg='BLASpp requires a threadsafe openblas') + conflicts('^openblas@0.3.6 threads=none', msg='BLAS++ requires a threadsafe openblas') + conflicts('^openblas@0.3.7: ~locking', msg='BLAS++ requires a threadsafe openblas') + + conflicts('+rocm', when='@:2020.10.02', msg='ROCm support requires BLAS++ 2021.04.00 or greater') + conflicts('+rocm', when='+cuda', msg='BLAS++ can only support one GPU backend at a time') def cmake_args(self): spec = self.spec + backend_config = '-Duse_cuda=%s' % ('+cuda' in spec) + if self.version >= Version('2021.04.01'): + backend = 'none' + if '+cuda' in spec: + backend = 'cuda' + if '+rocm' in spec: + backend = 'hip' + backend_config = '-Dgpu_backend=%s' % backend return [ '-Dbuild_tests=%s' % self.run_tests, '-Duse_openmp=%s' % ('+openmp' in spec), '-DBUILD_SHARED_LIBS=%s' % ('+shared' in spec), - '-Duse_cuda=%s' % ('+cuda' in spec), + backend_config, '-DBLAS_LIBRARIES=%s' % spec['blas'].libs.joined(';') ] def check(self): # If the tester fails to build, ensure that the check() fails. if os.path.isfile(join_path(self.build_directory, 'test', 'tester')): - make('check') + with working_dir(self.build_directory): + make('check') else: raise Exception('The tester was not built!') |