Skip to content

Latest commit

 

History

History
74 lines (45 loc) · 783 Bytes

api.rst

File metadata and controls

74 lines (45 loc) · 783 Bytes
Raw

API Reference

Utilities

has_cuda_support

.. autofunction:: mpi4jax.has_cuda_support

Communication primitives

allgather

.. autofunction:: mpi4jax.allgather

allreduce

.. autofunction:: mpi4jax.allreduce

alltoall

.. autofunction:: mpi4jax.alltoall

barrier

.. autofunction:: mpi4jax.barrier

bcast

.. autofunction:: mpi4jax.bcast

gather

.. autofunction:: mpi4jax.gather

recv

.. autofunction:: mpi4jax.recv

reduce

.. autofunction:: mpi4jax.reduce

scan

.. autofunction:: mpi4jax.scan

scatter

.. autofunction:: mpi4jax.scatter

send

.. autofunction:: mpi4jax.send

sendrecv

.. autofunction:: mpi4jax.sendrecv