# IT4Innovations
# LK 2024

name = 'OpenMPI'
version = '5.0.5'
versionsuffix = '-acfl-24.10'

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

toolchain = {'name': 'GCC', 'version': '14.2.0'}

source_urls = ['https://www.open-mpi.org/software/ompi/v%(version_major_minor)s/downloads']
sources = [SOURCELOWER_TAR_BZ2]
checksums = [
    {'openmpi-5.0.5.tar.bz2': '6588d57c0a4bd299a24103f4e196051b29e8b55fbda49e11d5b3d32030a32776'},
]

builddependencies = [
    ('pkgconf', '2.2.0'),
    ('Perl', '5.38.2'),
    ('Autotools', '20231222'),
]

dependencies = [
    ('acfl', '24.10', '', True),
    ('armpl', '24.10.0-RHEL-arm-linux-compiler', '', True),
    ('zlib', '1.3.1'),
    ('hwloc', '2.10.0'),
    ('libevent', '2.1.12'),
    ('UCX', '1.16.0'),
    ('libfabric', '1.16.1'),
    ('PMIx', '5.0.2'),
    ('UCC', '1.3.0'),
    ('PRRTE', '3.0.5'),
]

# Update configure to include changes from the "internal-cuda" patch
# by running a subset of autogen.pl sufficient to achieve this
# without doing the full, long-running regeneration.
#preconfigopts = ' && '.join([
#    'cd config',
#    'autom4te --language=m4sh opal_get_version.m4sh -o opal_get_version.sh',
#    'cd ..',
#    'autoconf',
#    'autoheader',
#    'aclocal',
#    'automake',
#    ''
#])

preconfigopts = 'CC=armclang CXX=armclang++ FC=armflang '

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

import os
if os.environ.get("CLUSTERNAME") in ["BARBORA"]:
   modextravars = {'OMPI_MCA_btl_openib_if_include': 'mlx5_0',
                'OMPI_MCA_btl_tcp_if_include': '10.33.4.0/24',
                'OMPI_MCA_orte_base_help_aggregate': '0',
                'SLURM_MPI_TYPE': 'pmix_v4',
   }
elif os.environ.get("CLUSTERNAME") in ["KAROLINA"]:
   modextravars = {'OMPI_MCA_btl_openib_if_include': 'mlx5_0',
                'OMPI_MCA_orte_base_help_aggregate': '0',
                'SLURM_MPI_TYPE': 'pmix_v4',
   }
else:
   modextravars = {'OMPI_MCA_btl_openib_if_include': 'mlx4_0',
                'OMPI_MCA_oob_tcp_if_include': '10.0.0.0/8',
                'SLURM_MPI_TYPE': 'pmix_v4',
   }


moduleclass = 'mpi'