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

mpich: Add CUDA variant (#29901)

Browse Source 
CUDA device memory is supported with the ch4 device.

Update provides to include MPI-4.0 functionality
This commit is contained in:
Ken Raffenetti 2022-04-22 10:02:27 -05:00 committed by GitHub
parent 960b48b613
commit a6a7c3ce88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
var/spack/repos/builtin/packages/mpich/package.py
View File

@ -10,7 +10,7 @@
from spack import * from spack import *
class Mpich(AutotoolsPackage): class Mpich(AutotoolsPackage, CudaPackage):
"""MPICH is a high performance and widely portable implementation of """MPICH is a high performance and widely portable implementation of
the Message Passing Interface (MPI) standard.""" the Message Passing Interface (MPI) standard."""
@ -94,7 +94,13 @@ class Mpich(AutotoolsPackage):
'-Wl,-flat_namespace'.''' '-Wl,-flat_namespace'.'''
) )
provides('mpi@:3.1') # Todo: cuda can be a conditional variant, but it does not seem to work when
# overriding the variant from CudaPackage.
conflicts('+cuda', when='@:3.3')
conflicts('+cuda', when='device=ch3')
provides('mpi@:4.0')
provides('mpi@:3.1', when='@:3.2')
provides('mpi@:3.0', when='@:3.1') provides('mpi@:3.0', when='@:3.1')
provides('mpi@:2.2', when='@:1.2') provides('mpi@:2.2', when='@:1.2')
provides('mpi@:2.1', when='@:1.1') provides('mpi@:2.1', when='@:1.1')
@ -433,7 +439,6 @@ def die_without_fortran(self):
def configure_args(self): def configure_args(self):
spec = self.spec spec = self.spec
config_args = [ config_args = [
'--without-cuda',
'--disable-silent-rules', '--disable-silent-rules',
'--enable-shared', '--enable-shared',
'--with-hwloc-prefix={0}'.format( '--with-hwloc-prefix={0}'.format(
@ -468,6 +473,8 @@ def configure_args(self):
elif 'pmi=cray' in spec: elif 'pmi=cray' in spec:
config_args.append('--with-pmi=cray') config_args.append('--with-pmi=cray')
config_args += self.with_or_without('cuda', activation_value='prefix')
# setup device configuration # setup device configuration
device_config = '' device_config = ''
if 'device=ch4' in spec: if 'device=ch4' in spec: