Skip to content

compiler: Add support for IBM Spectrum MPI #2907

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
JDBetteridge merged 1 commit into from
Apr 30, 2026
Merged

compiler: Add support for IBM Spectrum MPI #2907

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
16 changes: 10 additions & 6 deletions devito/arch/compiler.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,8 @@ def sniff_mpi_distro(mpiexec):
return 'MPICH'
elif "Intel(R) MPI" in ver:
return 'IntelMPI'
elif "IBM Spectrum MPI" in ver:
return "SpectrumMPI"
except (CalledProcessError, UnicodeDecodeError):
pass
return 'unknown'
Expand All @@ -124,14 +126,16 @@ def sniff_mpi_distro(mpiexec):
@memoized_func
def sniff_mpi_flags(mpicc='mpicc'):
mpi_distro = sniff_mpi_distro('mpiexec')
if mpi_distro != 'OpenMPI':
if mpi_distro in ['OpenMPI', 'SpectrumMPI']:
# OpenMPI's CC wrapper, namely mpicc, takes the --showme argument to find out
# the flags used for compiling and linking
compile_flags = check_output(['mpicc', "--showme:compile"]).decode("utf-8")
link_flags = check_output(['mpicc', "--showme:link"]).decode("utf-8")
else:
# TODO: This can be obtained from MPICH `mpicc -show`
# but does not segregate compile and link flags
raise NotImplementedError("Unable to detect MPI compile and link flags")

# OpenMPI's CC wrapper, namely mpicc, takes the --showme argument to find out
# the flags used for compiling and linking
compile_flags = check_output(['mpicc', "--showme:compile"]).decode("utf-8")
link_flags = check_output(['mpicc', "--showme:link"]).decode("utf-8")

return compile_flags.split(), link_flags.split()


Expand Down
Loading