Enzyme
Enzyme copied to clipboard
Handle more MPI functions
I am looking to use Enzyme on the MOM6 codebase. MOM6 depends on the library FMS which makes use of MPI.
There are several MPI function in FMS that don't appear to be handled in Enzyme:
- mpi_alltoall
- mpi_alltoallv
- mpi_alltoallw
- mpi_comm_group
- mpi_group_incl
- mpi_info_create
- mpi_info_set
- mpi_initialized
- mpi_type_commit
- mpi_type_create_subarray
- mpi_type_free
- mpi_wtick
Also it looks like work on a couple of these has already been start in PR #925
Are you making use of MPI_Op for reductions? Would also be good to check for those and reference any missing ops here.
Following @tgymnich suggestion, I've had a look at FMS and it also uses MPI_MIN, and MPI_MAX reductions, though these are only used with MPI_Allreduce.