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

Jump to bottom

Lower MPI to `stablehlo.custom_call` ops #305

Draft

mofeing wants to merge 1 commit into main from ss/mpi-to-stablehlo

+338 −0

Collaborator

mofeing commented Feb 2, 2025 •

edited

Loading

notes

MPI types are not ABI stable, but fortunately there is MPItrampoline which provides some wrappers over them
- MPIABI_Comm
- MPIABI_Request
- MPIABI_Op
docs on how to register and use a custom_call:
- https://jax.readthedocs.io/en/latest/ffi.html#c-interface
- https://openxla.org/xla/custom_call

questions

should we build libmpitrampoline into it or should we link on runtime?
how do we register stablehlo.custom_calls? do we register them on some initialization phase in Reactant.jl or here?


          Prototype conversion

02bccbe

mofeing marked this pull request as draft

February 2, 2025 12:45

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet