Enzyme icon indicating copy to clipboard operation
Enzyme copied to clipboard

Handle more MPI functions

Open l90lpa opened this issue 2 years ago • 3 comments

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

l90lpa avatar Mar 27 '23 18:03 l90lpa

Also it looks like work on a couple of these has already been start in PR #925

l90lpa avatar Mar 27 '23 18:03 l90lpa

Are you making use of MPI_Op for reductions? Would also be good to check for those and reference any missing ops here.

tgymnich avatar Mar 27 '23 19:03 tgymnich

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.

l90lpa avatar Mar 28 '23 14:03 l90lpa