Skip to content

Add oneapi-ifx to MPI_COMPILERS #314

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Oct 25, 2025
Merged

Add oneapi-ifx to MPI_COMPILERS #314

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 15 additions & 13 deletions mache/spack/env.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,14 @@
from mache.spack.shared import _get_modules, _get_yaml_data
from mache.version import __version__

MPI_COMPILERS = {
'gnu': {'mpicc': 'mpicc', 'mpicxx': 'mpicxx', 'mpifc': 'mpif90'},
'intel': {'mpicc': 'mpicc', 'mpicxx': 'mpicxx', 'mpifc': 'mpif90'},
'oneapi-ifx': {'mpicc': 'mpicc', 'mpicxx': 'mpicxx', 'mpifc': 'mpif90'},
'impi': {'mpicc': 'mpiicc', 'mpicxx': 'mpiicpc', 'mpifc': 'mpiifort'},
'cray': {'mpicc': 'cc', 'mpicxx': 'CC', 'mpifc': 'ftn'},
}


def make_spack_env(
spack_path,
Expand Down Expand Up @@ -287,29 +295,23 @@ def get_modules_env_vars_and_mpi_compilers(
def _get_mpi_compilers(machine, compiler, mpi, cray_compilers):
"""Get a list of compilers from a yaml file"""

mpi_compilers = {
'gnu': {'mpicc': 'mpicc', 'mpicxx': 'mpicxx', 'mpifc': 'mpif90'},
'intel': {'mpicc': 'mpicc', 'mpicxx': 'mpicxx', 'mpifc': 'mpif90'},
'impi': {'mpicc': 'mpiicc', 'mpicxx': 'mpiicpc', 'mpifc': 'mpiifort'},
'cray': {'mpicc': 'cc', 'mpicxx': 'CC', 'mpifc': 'ftn'},
}

mpi_compiler = None
# first, get mpi compilers based on compiler
if compiler in mpi_compilers:
mpi_compiler = mpi_compilers[compiler]
if compiler in MPI_COMPILERS:
mpi_compiler = MPI_COMPILERS[compiler]

# next, get mpi compilers based on mpi (higher priority)
if mpi in mpi_compilers:
mpi_compiler = mpi_compilers[mpi]
if mpi in MPI_COMPILERS:
mpi_compiler = MPI_COMPILERS[mpi]

# finally, get mpi compilers if this is a cray machine (highest priority)
if cray_compilers:
mpi_compiler = mpi_compilers['cray']
mpi_compiler = MPI_COMPILERS['cray']

if mpi_compiler is None:
raise ValueError(
f"Couldn't figure out MPI compilers for {machine} {compiler} {mpi}"
f"Couldn't figure out MPI compiler wrappers for {machine} "
f'{compiler} {mpi}'
)

return mpi_compiler['mpicc'], mpi_compiler['mpicxx'], mpi_compiler['mpifc']
Loading