execute_nccl Subroutine

private subroutine execute_nccl(self, in, out, stream, aux, error_code)

Uses

  • proc~~execute_nccl~~UsesGraph proc~execute_nccl backend_nccl%execute_nccl iso_c_binding iso_c_binding proc~execute_nccl->iso_c_binding iso_fortran_env iso_fortran_env proc~execute_nccl->iso_fortran_env

Executes NCCL backend

Type Bound

backend_nccl

Arguments

Type IntentOptional Attributes Name
class(backend_nccl), intent(inout) :: self

NCCL backend
real(kind=real32), intent(inout), target :: in(:)

Send pointer
real(kind=real32), intent(inout), target :: out(:)

Recv pointer
type(dtfft_stream_t), intent(in) :: stream

Main execution CUDA stream
real(kind=real32), intent(inout), target :: aux(:)

Aux pointer
integer(kind=int32), intent(out) :: error_code

Error code

Calls

proc~~execute_nccl~~CallsGraph proc~execute_nccl backend_nccl%execute_nccl fname fname proc~execute_nccl->fname interface~ncclgroupend ncclGroupEnd proc~execute_nccl->interface~ncclgroupend interface~ncclgroupstart ncclGroupStart proc~execute_nccl->interface~ncclgroupstart interface~ncclrecv ncclRecv proc~execute_nccl->interface~ncclrecv interface~ncclsend ncclSend proc~execute_nccl->interface~ncclsend mpi_abort mpi_abort proc~execute_nccl->mpi_abort proc~execute~2 abstract_kernel%execute proc~execute_nccl->proc~execute~2 proc~ncclgeterrorstring ncclGetErrorString proc~execute_nccl->proc~ncclgeterrorstring execute_private execute_private proc~execute~2->execute_private proc~pop_nvtx_domain_range pop_nvtx_domain_range proc~execute~2->proc~pop_nvtx_domain_range proc~push_nvtx_domain_range push_nvtx_domain_range proc~execute~2->proc~push_nvtx_domain_range interface~ncclgeterrorstring_c ncclGetErrorString_c proc~ncclgeterrorstring->interface~ncclgeterrorstring_c proc~string_c2f string_c2f proc~ncclgeterrorstring->proc~string_c2f interface~nvtxdomainrangepop_c nvtxDomainRangePop_c proc~pop_nvtx_domain_range->interface~nvtxdomainrangepop_c interface~nvtxdomainrangepushex_c nvtxDomainRangePushEx_c proc~push_nvtx_domain_range->interface~nvtxdomainrangepushex_c proc~astring_f2c astring_f2c proc~push_nvtx_domain_range->proc~astring_f2c proc~create_nvtx_domain create_nvtx_domain proc~push_nvtx_domain_range->proc~create_nvtx_domain interface~is_null_ptr is_null_ptr proc~string_c2f->interface~is_null_ptr interface~is_null_ptr->interface~is_null_ptr proc~is_null_funptr is_null_funptr interface~is_null_ptr->proc~is_null_funptr proc~string_f2c string_f2c proc~astring_f2c->proc~string_f2c proc~create_nvtx_domain->proc~astring_f2c interface~nvtxdomaincreate_c nvtxDomainCreate_c proc~create_nvtx_domain->interface~nvtxdomaincreate_c