dbcsr_acc_operations Module

Accelerator support for DBCSR



Contents


Variables

TypeVisibilityAttributesNameInitial
logical, private, parameter:: careful_mod =.FALSE.

Subroutines

public subroutine dbcsr_acc_do_mm_stack(param_stack_host, param_stack_dev, stack_size, data_type, a_data, b_data, c_data, m_max, n_max, k_max, def_mnk, stack_stream, c_stream, success)

Launch an accelerated kernel for processing a stack.

Arguments

TypeIntentOptionalAttributesName
integer, intent(in), DIMENSION(:, :), TARGET:: param_stack_host
type(acc_devmem_type), intent(in) :: param_stack_dev
integer, intent(in) :: stack_size
integer, intent(in) :: data_type
type(acc_devmem_type), intent(in) :: a_data
type(acc_devmem_type), intent(in) :: b_data
type(acc_devmem_type), intent(inout) :: c_data
integer, intent(in) :: m_max
integer, intent(in) :: n_max
integer, intent(in) :: k_max
logical, intent(in) :: def_mnk
type(acc_stream_type), intent(in) :: stack_stream
type(acc_stream_type), intent(in) :: c_stream
logical, intent(inout) :: success

public subroutine dbcsr_acc_transpose(trs_stack, offset, nblks, data_type, buffer, m, n, stream)

Launch an accelerated transpose kernel

Arguments

TypeIntentOptionalAttributesName
type(acc_devmem_type), intent(in) :: trs_stack
integer, intent(in) :: offset
integer, intent(in) :: nblks
integer, intent(in) :: data_type
type(acc_devmem_type), intent(in) :: buffer
integer, intent(in) :: m
integer, intent(in) :: n
type(acc_stream_type), intent(in) :: stream