Added new package: spla (#17868)
This commit is contained in:
parent
4eb3558d20
commit
cb676eab0f
47
var/spack/repos/builtin/packages/spla/package.py
Normal file
47
var/spack/repos/builtin/packages/spla/package.py
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
# 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 Spla(CMakePackage):
|
||||||
|
"""Specialized Parallel Linear Algebra, providing distributed GEMM
|
||||||
|
functionality for specific matrix distributions with optional GPU
|
||||||
|
acceleration."""
|
||||||
|
|
||||||
|
homepage = "https://github.com/eth-cscs/spla"
|
||||||
|
url = "https://github.com/eth-cscs/spla/archive/v1.0.0.tar.gz"
|
||||||
|
git = 'https://github.com/eth-cscs/spla.git'
|
||||||
|
|
||||||
|
version('1.0.0', sha256='a0eb269b84d7525b223dc650de12170bba30fbb3ae4f93eb2b5cbdce335e4da1')
|
||||||
|
version('master', branch='master')
|
||||||
|
|
||||||
|
variant('openmp', default=True, description="Build with OpenMP support")
|
||||||
|
variant('static', default=False, description="Build as static library")
|
||||||
|
variant('cuda', default=False, description="CUDA")
|
||||||
|
|
||||||
|
depends_on('mpi')
|
||||||
|
depends_on('blas')
|
||||||
|
depends_on('cuda', when='+cuda')
|
||||||
|
depends_on('cmake@3.10:', type='build')
|
||||||
|
|
||||||
|
def cmake_args(self):
|
||||||
|
args = []
|
||||||
|
|
||||||
|
if '+openmp' in self.spec:
|
||||||
|
args += ["-DSPLA_OMP=ON"]
|
||||||
|
else:
|
||||||
|
args += ["-DSPLA_OMP=OFF"]
|
||||||
|
|
||||||
|
if '+cuda' in self.spec:
|
||||||
|
args += ["-DSPLA_GPU_BACKEND=CUDA"]
|
||||||
|
else:
|
||||||
|
args += ["-DSPLA_GPU_BACKEND=OFF"]
|
||||||
|
|
||||||
|
if '+static' in self.spec:
|
||||||
|
args += ["-DSPLA_STATIC=ON"]
|
||||||
|
else:
|
||||||
|
args += ["-DSPLA_STATIC=OFF"]
|
||||||
|
return args
|
Loading…
Reference in New Issue
Block a user