Accelerator support for DBCSR
Type | Visibility | Attributes | Name | Initial | |||
---|---|---|---|---|---|---|---|
logical, | private, | parameter | :: | careful_mod | = | .FALSE. |
Launch an accelerated kernel for processing a stack.
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 | |||
logical, | intent(inout) | :: | generated_acc_untuned |
Launch an accelerated transpose kernel
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 |