dbcsr_acc_operations Module

Accelerator support for DBCSR



Variables

Type Visibility Attributes Name Initial
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, generated_acc_untuned)

Launch an accelerated kernel for processing a stack.

Arguments

Type IntentOptional Attributes Name
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
logical, intent(inout) :: generated_acc_untuned

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

Launch an accelerated transpose kernel

Arguments

Type IntentOptional Attributes Name
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