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)

Launch an accelerated kernel for processing a stack.

Arguments

Type	Intent	Optional	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

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

Launch an accelerated transpose kernel

Arguments

Type	Intent	Optional	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