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 |