From c68295df2c0d45eaf79a882c81bd06b5b663e5a5 Mon Sep 17 00:00:00 2001 From: 健美猫 Date: Fri, 24 Mar 2017 20:48:08 +0800 Subject: Add cuda support for gromacs. (#3527) --- var/spack/repos/builtin/packages/gromacs/package.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/var/spack/repos/builtin/packages/gromacs/package.py b/var/spack/repos/builtin/packages/gromacs/package.py index fd5814ca08..40d7858816 100644 --- a/var/spack/repos/builtin/packages/gromacs/package.py +++ b/var/spack/repos/builtin/packages/gromacs/package.py @@ -51,14 +51,14 @@ class Gromacs(CMakePackage): 'double', default=False, description='Produces a double precision version of the executables') variant('plumed', default=False, description='Enable PLUMED support') + variant('cuda', default=False, description='Enable CUDA support') depends_on('mpi', when='+mpi') depends_on('plumed+mpi', when='+plumed+mpi') depends_on('plumed~mpi', when='+plumed~mpi') depends_on('fftw') depends_on('cmake@2.8.8:', type='build') - - # TODO : add GPU support + depends_on('cuda', when='+cuda') def patch(self): if '+plumed' in self.spec: @@ -82,4 +82,9 @@ class Gromacs(CMakePackage): else: options.append('-DCMAKE_BUILD_TYPE:STRING=Release') + if '+cuda' in self.spec: + options.append('-DGMX_GPU:BOOL=ON') + options.append('-DCUDA_TOOLKIT_ROOT_DIR:STRING=' + + self.spec['cuda'].prefix) + return options -- cgit v1.2.3-70-g09d2