alloc_and_set_aux Function

private function alloc_and_set_aux(helper, backend, cart_comm, aux, paux, plans, bplans) result(is_aux_alloc)

Uses

  • proc~~alloc_and_set_aux~~UsesGraph proc~alloc_and_set_aux alloc_and_set_aux iso_fortran_env iso_fortran_env proc~alloc_and_set_aux->iso_fortran_env

Allocates auxiliary memory according to the backend and sets it to the plans

Arguments

Type IntentOptional Attributes Name
type(backend_helper), intent(inout) :: helper

Backend helper
type(dtfft_backend_t), intent(in) :: backend

GPU backend
type(MPI_Comm), intent(in) :: cart_comm

Cartesian communicator
type(c_ptr), intent(inout) :: aux

Allocatable auxiliary memory
real(kind=real32), intent(inout), pointer :: paux(:)

Pointer to auxiliary memory
type(transpose_handle_cuda), intent(inout) :: plans(:)

Plans
type(transpose_handle_cuda), intent(inout), optional :: bplans(:)

Backward plans

Return Value logical

Is auxiliary memory allocated


Calls

proc~~alloc_and_set_aux~~CallsGraph proc~alloc_and_set_aux alloc_and_set_aux mpi_abort mpi_abort proc~alloc_and_set_aux->mpi_abort mpi_allreduce mpi_allreduce proc~alloc_and_set_aux->mpi_allreduce proc~alloc_mem alloc_mem proc~alloc_and_set_aux->proc~alloc_mem proc~dtfft_get_error_string dtfft_get_error_string proc~alloc_and_set_aux->proc~dtfft_get_error_string proc~get_aux_size~2 transpose_handle_cuda%get_aux_size proc~alloc_and_set_aux->proc~get_aux_size~2 proc~alloc_mem->mpi_abort proc~alloc_mem->mpi_allreduce interface~cudamalloc cudaMalloc proc~alloc_mem->interface~cudamalloc interface~cudamemgetinfo cudaMemGetInfo proc~alloc_mem->interface~cudamemgetinfo interface~int_to_str int_to_str proc~alloc_mem->interface~int_to_str interface~ncclcommregister ncclCommRegister proc~alloc_mem->interface~ncclcommregister interface~ncclmemalloc ncclMemAlloc proc~alloc_mem->interface~ncclmemalloc interface~nvshmem_malloc nvshmem_malloc proc~alloc_mem->interface~nvshmem_malloc is_null_ptr is_null_ptr proc~alloc_mem->is_null_ptr proc~cudageterrorstring cudaGetErrorString proc~alloc_mem->proc~cudageterrorstring proc~dtfft_get_backend_string dtfft_get_backend_string proc~alloc_mem->proc~dtfft_get_backend_string proc~get_log_enabled get_log_enabled proc~alloc_mem->proc~get_log_enabled proc~is_backend_nccl is_backend_nccl proc~alloc_mem->proc~is_backend_nccl proc~is_backend_nvshmem is_backend_nvshmem proc~alloc_mem->proc~is_backend_nvshmem proc~ncclgeterrorstring ncclGetErrorString proc~alloc_mem->proc~ncclgeterrorstring proc~write_message write_message proc~alloc_mem->proc~write_message temp temp proc~alloc_mem->temp proc~get_aux_size abstract_backend%get_aux_size proc~get_aux_size~2->proc~get_aux_size proc~int_to_str_int32 int_to_str_int32 interface~int_to_str->proc~int_to_str_int32 proc~int_to_str_int64 int_to_str_int64 interface~int_to_str->proc~int_to_str_int64 proc~int_to_str_int8 int_to_str_int8 interface~int_to_str->proc~int_to_str_int8 interface~cudageterrorstring_c cudaGetErrorString_c proc~cudageterrorstring->interface~cudageterrorstring_c proc~string_c2f string_c2f proc~cudageterrorstring->proc~string_c2f interface~ncclgeterrorstring_c ncclGetErrorString_c proc~ncclgeterrorstring->interface~ncclgeterrorstring_c proc~ncclgeterrorstring->proc~string_c2f mpi_comm_rank mpi_comm_rank proc~write_message->mpi_comm_rank mpi_finalized mpi_finalized proc~write_message->mpi_finalized

Called by

proc~~alloc_and_set_aux~~CalledByGraph proc~alloc_and_set_aux alloc_and_set_aux proc~create_cuda transpose_plan_cuda%create_cuda proc~create_cuda->proc~alloc_and_set_aux proc~run_autotune_backend run_autotune_backend proc~create_cuda->proc~run_autotune_backend proc~autotune_grid_decomposition autotune_grid_decomposition proc~create_cuda->proc~autotune_grid_decomposition proc~run_autotune_backend->proc~alloc_and_set_aux proc~autotune_grid autotune_grid proc~autotune_grid->proc~run_autotune_backend proc~autotune_grid_decomposition->proc~autotune_grid