zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Relion 2.1 - Using CudaPackage class (#6707)

Browse Source
This commit is contained in:
Douglas Duckworth 2018-03-07 01:45:48 -05:00 committed by Massimiliano Culpo
parent 95b2738549
commit 0f114d9718
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
var/spack/repos/builtin/packages/relion/package.py
View File

@ -25,7 +25,7 @@
from spack import * from spack import *
class Relion(CMakePackage): class Relion(CMakePackage, CudaPackage):
"""RELION (for REgularised LIkelihood OptimisatioN, pronounce rely-on) is a """RELION (for REgularised LIkelihood OptimisatioN, pronounce rely-on) is a
stand-alone computer program that employs an empirical Bayesian approach to stand-alone computer program that employs an empirical Bayesian approach to
refinement of (multiple) 3D reconstructions or 2D class averages in refinement of (multiple) 3D reconstructions or 2D class averages in
@ -40,10 +40,6 @@ class Relion(CMakePackage):
variant('gui', default=True, description="build the gui") variant('gui', default=True, description="build the gui")
variant('cuda', default=True, description="enable compute on gpu") variant('cuda', default=True, description="enable compute on gpu")
variant('cuda_arch', default=None, description='CUDA architecture',
values=('20', '30', '32', '35', '50', '52', '53', '60', '61', '62'
'70'),
multi=True)
variant('double', default=True, description="double precision (cpu) code") variant('double', default=True, description="double precision (cpu) code")
variant('double-gpu', default=False, description="double precision (gpu) code") variant('double-gpu', default=False, description="double precision (gpu) code")
variant('build_type', default='RelWithDebInfo', variant('build_type', default='RelWithDebInfo',
@ -68,15 +64,15 @@ def cmake_args(self):
'-DDoublePrec_CPU=%s' % ('+double' in self.spec), '-DDoublePrec_CPU=%s' % ('+double' in self.spec),
'-DDoublePrec_GPU=%s' % ('+double-gpu' in self.spec), '-DDoublePrec_GPU=%s' % ('+double-gpu' in self.spec),
] ]
carch = self.spec.variants['cuda_arch'].value
if '+cuda' in self.spec: if '+cuda' in self.spec:
args += [ args += [
'-DCUDA=on', '-DCUDA=on',
] ]
if carch is not None:
carch = self.spec.variants['cuda_arch'].value args += [
'-DCUDA_ARCH=%s' % (carch),
if carch is not None: ]
args += [
'-DCUDA_ARCH=%s' % (carch),
]
return args return args