gs_device_nccl.F90 File Reference

Go to the source code of this file.

Data Types
type	gs_device_nccl::gs_device_nccl_buf_t
	Buffers for non-blocking communication and packing/unpacking. More...
type	gs_device_nccl::gs_device_nccl_t
	Gather-scatter communication using NCCL The arrays are indexed per PE like send_pe and @ recv_pe. More...
interface	gs_device_nccl::hip_gs_pack
interface	gs_device_nccl::hip_gs_unpack
interface	gs_device_nccl::device_nccl_sendrecv

Modules
module	gs_device_nccl
	Defines NCCL based gather-scatter communication.

Functions/Subroutines
subroutine	gs_device_nccl::gs_device_nccl_buf_init (this, pe_order, dof_stack, mark_dupes)
subroutine	gs_device_nccl::gs_device_nccl_buf_free (this)
subroutine	gs_device_nccl::gs_device_nccl_init (this, send_pe, recv_pe)
	Initialise NCCL based communication method.
subroutine	gs_device_nccl::gs_device_nccl_free (this)
	Deallocate MPI based communication method.
subroutine	gs_device_nccl::gs_device_nccl_nbsend (this, u, n, deps, strm)
	Post non-blocking send operations.
subroutine	gs_device_nccl::gs_device_nccl_nbrecv (this)
	Post non-blocking receive operations.
subroutine	gs_device_nccl::gs_device_nccl_nbwait (this, u, n, op, strm)
	Wait for non-blocking operations.