nvrtc_kernel Derived Type

type, public :: nvrtc_kernel

nvRTC Compiled kernel class


Inherits

type~~nvrtc_kernel~~InheritsGraph type~nvrtc_kernel nvrtc_kernel type~cufunction CUfunction type~nvrtc_kernel->type~cufunction cuda_kernel type~dim3 dim3 type~nvrtc_kernel->type~dim3 num_blocks, block_size type~kernelargs kernelArgs type~nvrtc_kernel->type~kernelargs args c_ptr c_ptr type~cufunction->c_ptr ptr type~kernelargs->c_ptr ptrs

Inherited by

type~~nvrtc_kernel~~InheritedByGraph type~nvrtc_kernel nvrtc_kernel type~abstract_backend abstract_backend type~abstract_backend->type~nvrtc_kernel unpack_kernel, unpack_kernel2 type~transpose_handle_cuda transpose_handle_cuda type~transpose_handle_cuda->type~nvrtc_kernel transpose_kernel, unpack_kernel, unpack_kernel2 type~transpose_handle_cuda->type~abstract_backend comm_handle type~backend_cufftmp backend_cufftmp type~backend_cufftmp->type~abstract_backend type~backend_mpi backend_mpi type~backend_mpi->type~abstract_backend type~backend_nccl backend_nccl type~backend_nccl->type~abstract_backend type~transpose_plan_cuda transpose_plan_cuda type~transpose_plan_cuda->type~transpose_handle_cuda fplans, bplans

Components

Type Visibility Attributes Name Initial
logical, private :: is_created = .false.

Kernel is created flag.
logical, private :: is_dummy = .false.

If kernel should do anything or not.
type(CUfunction), private :: cuda_kernel

Pointer to CUDA kernel.
type(dim3), private :: num_blocks

Grid of blocks.
type(dim3), private :: block_size

Thread block.
integer(kind=int8), private :: kernel_type

Type of kernel to execute.
type(kernelArgs), private :: args

Kernel arguments.
integer(kind=int32), private, allocatable :: pointers(:,:)

Optional pointers that hold info about counts and displacements in KERNEL_UNPACK_PIPELINED kernel.

Type-Bound Procedures

procedure, public, pass(self) :: create

Creates kernel

  • private subroutine create(self, comm, dims, base_storage, transpose_type, kernel_type, pointers)

    Creates kernel

    Arguments

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

    nvRTC Compiled kernel class
    type(MPI_Comm), intent(in) :: comm

    MPI Communicator
    integer(kind=int32), intent(in), target :: dims(0:)

    Global dimensions to process
    integer(kind=int64), intent(in) :: base_storage

    Number of bytes needed to store single element
    type(dtfft_transpose_t), intent(in) :: transpose_type

    Type of transposition to perform
    integer(kind=int8), intent(in) :: kernel_type

    Type of kernel to build
    integer(kind=int32), intent(in), optional :: pointers(:,:)

    Optional pointers to unpack kernels

procedure, public, pass(self) :: execute

Executes kernel

  • private subroutine execute(self, in, out, stream, source)

    Executes kernel on stream

    Arguments

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

    nvRTC Compiled kernel class
    real(kind=real32), intent(in), target :: in(:)

    Source pointer
    real(kind=real32), intent(in), target :: out(:)

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

    CUDA Stream
    integer(kind=int32), intent(in), optional :: source

    Source rank for pipelined unpacking

procedure, public, pass(self) :: destroy

Destroys kernel

  • private subroutine destroy(self)

    Destroys kernel

    Arguments

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

    nvRTC Compiled kernel class