From 7d0ae0f2958c5f4e0301e041686ddb8c4df3628b Mon Sep 17 00:00:00 2001 From: Simon Pintarelli <1237199+simonpintarelli@users.noreply.github.com> Date: Tue, 22 Sep 2020 17:35:11 +0200 Subject: sirius: use -DCUDA_ARCH for develop, version >7.0.0 (#18852) Also remove master branch --- var/spack/repos/builtin/packages/sirius/package.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/var/spack/repos/builtin/packages/sirius/package.py b/var/spack/repos/builtin/packages/sirius/package.py index d4d8abfbac..f100fc9d0e 100644 --- a/var/spack/repos/builtin/packages/sirius/package.py +++ b/var/spack/repos/builtin/packages/sirius/package.py @@ -19,7 +19,6 @@ class Sirius(CMakePackage, CudaPackage): maintainers = ['simonpintarelli', 'haampie', 'dev-zero', 'AdhocMan'] version('develop', branch='develop') - version('master', branch='master') version('7.0.0', sha256='da783df11e7b65668e29ba8d55c8a6827e2216ad6d88040f84f42ac20fd1bb99') version('6.5.6', sha256='c8120100bde4477545eae489ea7f9140d264a3f88696ec92728616d78f214cae', @@ -82,13 +81,10 @@ class Sirius(CMakePackage, CudaPackage): depends_on('magma', when='+magma') depends_on('spfft', when='@6.4.0:') - depends_on('spfft', when='@master') depends_on('spfft', when='@develop') depends_on('spfft+cuda', when='@6.4.0:+cuda') - depends_on('spfft+cuda', when='@master+cuda') depends_on('spfft+cuda', when='@develop+cuda') depends_on('spfft+rocm', when='@6.4.0:+rocm') - depends_on('spfft+rocm', when='@master+rocm') depends_on('spfft+rocm', when='@develop+rocm') depends_on('spla@1.1.0:', when='@develop') @@ -205,12 +201,14 @@ class Sirius(CMakePackage, CudaPackage): ) args += ["-DELPA_INCLUDE_DIR={0}".format(elpa_incdir)] - if '+cuda' in spec: - cuda_arch = spec.variants['cuda_arch'].value - if cuda_arch[0] != 'none': - args += [ - '-DCMAKE_CUDA_FLAGS=-arch=sm_{0}'.format(cuda_arch[0]) - ] + cuda_arch = spec.variants['cuda_arch'].value + cuda_arch_selected = cuda_arch[0] != 'none' + if '+cuda' in spec and cuda_arch_selected: + cuda_args = '-DCUDA_ARCH={0}'.format(cuda_arch[0]) + if spec.satisfies('@:6.999'): + cuda_args = '-DCMAKE_CUDA_FLAGS=-arch=sm_{0}'.format( + cuda_arch[0]) + args.append(cuda_args) if '+rocm' in spec: archs = ",".join(self.spec.variants['amdgpu_target'].value) -- cgit v1.2.3-60-g2f50