# IT4Innovations 2021
# LK JK

name = 'OpenMPI'
version = '4.0.5'
versionsuffix = '-CUDA-11.2.2-test'

homepage = 'https://www.open-mpi.org/'
description = """The Open MPI Project is an open source MPI-3 implementation."""

toolchain = {'name': 'NVHPC', 'version': '21.2'}

source_urls = ['https://www.open-mpi.org/software/ompi/v%(version_major_minor)s/downloads']
sources = [SOURCELOWER_TAR_GZ]

dependencies = [
    ('UCX', '1.9.0', '-CUDA-11.2.2'),
    ('CUDAcore', '11.2.2', '', True),
    ('GDRCopy', '2.1', '-CUDA-11.2.2'),
]

preconfigopts = 'export CC="pgcc -noswitcherror" && export CXX="pgc++ -noswitcherror" && export FC="pgfortran -noswitcherror" && '

configopts = '--enable-shared --enable-mpi-thread-multiple --without-verbs '
configopts += '--with-hwloc=$EBROOTHWLOC '  # hwloc support
configopts += '--enable-mpirun-prefix-by-default '
configopts += '--with-slurm ' # Enable slurm
configopts += '--enable-mpi-cxx '  # Enable building the C++ MPI bindings
configopts += '--with-ucx=$EBROOTUCX '
configopts += '--with-cuda=$EBROOTCUDACORE '

moduleclass = 'mpi'