Loading [MathJax]/extensions/tex2jax.js
Neko 0.9.99
A portable framework for high-order spectral element flow simulations
All Classes Namespaces Files Functions Variables Typedefs Enumerator Macros Pages
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.