summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorTeodor Nikolov <teodor.nikolov22@gmail.com>2020-03-19 21:50:12 +0200
committerGitHub <noreply@github.com>2020-03-19 14:50:12 -0500
commita61c53a5c96c04d19116673e7c9d2bcc628c889e (patch)
treefafb11dae6c796a9bd1ea06bd3450c002a3f2508 /var
parentf9e58a52dad06caee5c7becd4e9025253a2571f4 (diff)
downloadspack-a61c53a5c96c04d19116673e7c9d2bcc628c889e.tar.gz
spack-a61c53a5c96c04d19116673e7c9d2bcc628c889e.tar.bz2
spack-a61c53a5c96c04d19116673e7c9d2bcc628c889e.tar.xz
spack-a61c53a5c96c04d19116673e7c9d2bcc628c889e.zip
cosma: added package at v2.0.2 (#15497)
* [new package] COSMA * Use virtual dependencies
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/cosma/package.py69
1 files changed, 69 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/cosma/package.py b/var/spack/repos/builtin/packages/cosma/package.py
new file mode 100644
index 0000000000..efb81035da
--- /dev/null
+++ b/var/spack/repos/builtin/packages/cosma/package.py
@@ -0,0 +1,69 @@
+# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+
+from spack import *
+
+
+class Cosma(CMakePackage):
+ """
+ Distributed Communication-Optimal Matrix-Matrix Multiplication Library
+ """
+
+ maintainers = ['teonnik', 'kabicm']
+ homepage = 'https://github.com/eth-cscs/COSMA'
+ url = 'https://github.com/eth-cscs/COSMA/releases/download/v2.0.2/cosma.tar.gz'
+ git = 'https://github.com/eth-cscs/COSMA.git'
+
+ # note: The default archives produced with github do not have the archives
+ # of the submodules.
+ version('master', branch='master', submodules=True)
+ version('2.0.2', sha256='4f3354828bc718f3eef2f0098c3bdca3499297497a220da32db1acd57920c68d')
+ # note: this version fails to build at the moment
+ # version('1.0.0',
+ # url='https://github.com/eth-cscs/COSMA/releases/download/1.0/cosma.tar.gz',
+ # sha256='c142104258dcca4c17fa7faffc2990a08d2777235c7980006e93c5dca51061f6')
+
+ variant('cuda', default=False,
+ description='Build with the CUBLAS back end.')
+ variant('scalapack', default=False,
+ description='Build with ScaLAPACK support.')
+
+ depends_on('cmake@3.12:', type='build')
+ depends_on('mpi@3:')
+ depends_on('blas', when='~cuda')
+ depends_on('scalapack', when='+scalapack')
+ # COSMA is written entirely in C++, it may use cublasXt but a CUDA capable
+ # compiler is not needed. There is no need for CudaPackage in this recipe.
+ depends_on('cuda', when='+cuda')
+
+ def setup_build_environment(self, env):
+ if '+cuda' in self.spec:
+ env.set('CUDA_PATH', self.spec['cuda'].prefix)
+
+ def cmake_args(self):
+ spec = self.spec
+ args = ['-DCOSMA_WITH_TESTS=OFF',
+ '-DCOSMA_WITH_APPS=OFF',
+ '-DCOSMA_WITH_PROFILING=OFF',
+ '-DCOSMA_WITH_BENCHMARKS=OFF']
+
+ if '^mkl' in spec:
+ args += ['-DCOSMA_BLAS=MKL']
+ elif '^netlib-lapack' in spec:
+ args += ['-DCOSMA_BLAS=CUSTOM']
+ elif '^openblas' in spec:
+ args += ['-DCOSMA_BLAS=OPENBLAS']
+ elif '+cuda' in spec:
+ args += ['-DCOSMA_BLAS=CUDA']
+ else: # TODO '^rocm' in spec:
+ args += ['-DCOSMA_BLAS=ROCM']
+
+ if '+scalapack' and '^mkl' in spec:
+ args += ['-DCOSMA_SCALAPACK=MKL']
+ elif '+scalapack' and '^netlib-scalapack' in spec:
+ args += ['-DCOSMA_SCALAPACK=CUSTOM']
+
+ return args