Neko  0.9.0
A portable framework for high-order spectral element flow simulations
gs_device_mpi Module Reference

Defines GPU aware MPI gather-scatter communication.

Data Types

type  gs_device_mpi_buf_t
 Buffers for non-blocking communication and packing/unpacking. More...
 
type  gs_device_mpi_t
 Gather-scatter communication using device MPI. The arrays are indexed per PE like send_pe and @ recv_pe. More...
 
interface  hip_gs_pack
 
interface  hip_gs_unpack
 
interface  device_mpi_init_reqs
 
interface  device_mpi_free_reqs
 
interface  device_mpi_isend
 
interface  device_mpi_irecv
 
interface  device_mpi_test
 
interface  device_mpi_waitall
 
interface  device_mpi_waitany
 

Functions/Subroutines

subroutine gs_device_mpi_buf_init (this, pe_order, dof_stack, mark_dupes)
 
subroutine gs_device_mpi_buf_free (this)
 
subroutine gs_device_mpi_init (this, send_pe, recv_pe)
 Initialise MPI based communication method. More...
 
subroutine gs_device_mpi_free (this)
 Deallocate MPI based communication method. More...
 
subroutine gs_device_mpi_nbsend (this, u, n, deps, strm)
 Post non-blocking send operations. More...
 
subroutine gs_device_mpi_nbrecv (this)
 Post non-blocking receive operations. More...
 
subroutine gs_device_mpi_nbwait (this, u, n, op, strm)
 Wait for non-blocking operations. More...
 

Function/Subroutine Documentation

◆ gs_device_mpi_buf_free()

subroutine gs_device_mpi::gs_device_mpi_buf_free ( class(gs_device_mpi_buf_t), intent(inout)  this)
private

Definition at line 261 of file gs_device_mpi.F90.

Here is the call graph for this function:

◆ gs_device_mpi_buf_init()

subroutine gs_device_mpi::gs_device_mpi_buf_init ( class(gs_device_mpi_buf_t), intent(inout)  this,
integer, dimension(:), intent(inout), allocatable  pe_order,
type(stack_i4_t), dimension(:), intent(inout), allocatable  dof_stack,
logical, intent(in)  mark_dupes 
)
private

Definition at line 191 of file gs_device_mpi.F90.

Here is the call graph for this function:

◆ gs_device_mpi_free()

subroutine gs_device_mpi::gs_device_mpi_free ( class(gs_device_mpi_t), intent(inout)  this)
private

Definition at line 304 of file gs_device_mpi.F90.

Here is the call graph for this function:

◆ gs_device_mpi_init()

subroutine gs_device_mpi::gs_device_mpi_init ( class(gs_device_mpi_t), intent(inout)  this,
type(stack_i4_t), intent(inout)  send_pe,
type(stack_i4_t), intent(inout)  recv_pe 
)
private

Definition at line 274 of file gs_device_mpi.F90.

Here is the call graph for this function:

◆ gs_device_mpi_nbrecv()

subroutine gs_device_mpi::gs_device_mpi_nbrecv ( class(gs_device_mpi_t), intent(inout)  this)
private

Definition at line 400 of file gs_device_mpi.F90.

◆ gs_device_mpi_nbsend()

subroutine gs_device_mpi::gs_device_mpi_nbsend ( class(gs_device_mpi_t), intent(inout)  this,
real(kind=rp), dimension(n), intent(inout)  u,
integer, intent(in)  n,
type(c_ptr), intent(inout)  deps,
type(c_ptr), intent(inout)  strm 
)
private

Definition at line 326 of file gs_device_mpi.F90.

Here is the call graph for this function:

◆ gs_device_mpi_nbwait()

subroutine gs_device_mpi::gs_device_mpi_nbwait ( class(gs_device_mpi_t), intent(inout)  this,
real(kind=rp), dimension(n), intent(inout)  u,
integer, intent(in)  n,
integer  op,
type(c_ptr), intent(inout)  strm 
)
private

Definition at line 413 of file gs_device_mpi.F90.

Here is the call graph for this function: