alloc_mem Subroutine

public subroutine alloc_mem(helper, backend, comm, alloc_bytes, ptr, error_code)

Uses

  • proc~~alloc_mem~~UsesGraph proc~alloc_mem alloc_mem iso_c_binding iso_c_binding proc~alloc_mem->iso_c_binding iso_fortran_env iso_fortran_env proc~alloc_mem->iso_fortran_env

Allocates memory based on backend

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) :: comm

MPI communicator
integer(kind=int64), intent(in) :: alloc_bytes

Number of bytes to allocate
type(c_ptr), intent(out) :: ptr

Pointer to the allocated memory
integer(kind=int32), intent(out) :: error_code

Error code

Calls

proc~~alloc_mem~~CallsGraph proc~alloc_mem alloc_mem 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 mpi_abort mpi_abort proc~alloc_mem->mpi_abort mpi_allreduce mpi_allreduce proc~alloc_mem->mpi_allreduce 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~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_mem~~CalledByGraph proc~alloc_mem alloc_mem proc~alloc_and_set_aux alloc_and_set_aux proc~alloc_and_set_aux->proc~alloc_mem proc~mem_alloc~3 abstract_transpose_plan%mem_alloc proc~mem_alloc~3->proc~alloc_mem proc~run_autotune_backend run_autotune_backend proc~run_autotune_backend->proc~alloc_mem proc~run_autotune_backend->proc~alloc_and_set_aux proc~autotune_grid autotune_grid proc~autotune_grid->proc~run_autotune_backend proc~create_cuda transpose_plan_cuda%create_cuda proc~create_cuda->proc~alloc_and_set_aux proc~create_cuda->proc~run_autotune_backend proc~autotune_grid_decomposition autotune_grid_decomposition proc~create_cuda->proc~autotune_grid_decomposition proc~autotune_grid_decomposition->proc~autotune_grid