dbcsr_tas_reshape_ops Module

communication routines to reshape / replicate / merge tall-and-skinny matrices.



Contents


Variables

TypeVisibilityAttributesNameInitial
character(len=*), private, parameter:: moduleN ='dbcsr_tas_reshape_ops'

Interfaces

private interface block_buffer_get_next_block

  • private subroutine block_buffer_get_next_area_block(buffer, ndata, index, block, advance_iter)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer, intent(out) :: ndata
    integer(kind=int_8), intent(out), DIMENSION(2):: index
    type(dbcsr_data_obj), intent(inout), optional :: block
    logical, intent(in), optional :: advance_iter

private interface block_buffer_add_block

  • private subroutine block_buffer_add_area_block(buffer, index, block, transposed)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer(kind=int_8), intent(in), DIMENSION(2):: index
    type(dbcsr_data_obj), intent(in) :: block
    logical, intent(in), optional :: transposed
  • private subroutine block_buffer_add_block_r_dp(buffer, index, block, transposed)

    insert a block into block buffer (at current iterator position)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer(kind=int_8), intent(in), DIMENSION(2):: index

    index of block

    real(kind=real_8), intent(in), DIMENSION(:, :):: block

    block

    logical, intent(in), optional :: transposed
  • private subroutine block_buffer_add_block_r_sp(buffer, index, block, transposed)

    insert a block into block buffer (at current iterator position)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer(kind=int_8), intent(in), DIMENSION(2):: index

    index of block

    real(kind=real_4), intent(in), DIMENSION(:, :):: block

    block

    logical, intent(in), optional :: transposed
  • private subroutine block_buffer_add_block_c_dp(buffer, index, block, transposed)

    insert a block into block buffer (at current iterator position)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer(kind=int_8), intent(in), DIMENSION(2):: index

    index of block

    complex(kind=real_8), intent(in), DIMENSION(:, :):: block

    block

    logical, intent(in), optional :: transposed
  • private subroutine block_buffer_add_block_c_sp(buffer, index, block, transposed)

    insert a block into block buffer (at current iterator position)

    Arguments

    TypeIntentOptionalAttributesName
    type(block_buffer_type), intent(inout) :: buffer
    integer(kind=int_8), intent(in), DIMENSION(2):: index

    index of block

    complex(kind=real_4), intent(in), DIMENSION(:, :):: block

    block

    logical, intent(in), optional :: transposed

Derived Types

type, private :: block_buffer_type

Components

TypeVisibilityAttributesNameInitial
integer, public :: nblock =-1
integer(kind=int_8), public, DIMENSION(:, :), ALLOCATABLE:: indx
real(kind=real_8), public, DIMENSION(:), ALLOCATABLE:: msg_r_dp
real(kind=real_4), public, DIMENSION(:), ALLOCATABLE:: msg_r_sp
complex(kind=real_8), public, DIMENSION(:), ALLOCATABLE:: msg_c_dp
complex(kind=real_4), public, DIMENSION(:), ALLOCATABLE:: msg_c_sp
integer, public :: data_type =-1
integer, public :: endpos =-1

Functions

private pure function block_buffer_blocks_left(buffer)

how many blocks left in iterator

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(in) :: buffer

Return Value logical


Subroutines

public recursive subroutine dbcsr_tas_reshape(matrix_in, matrix_out, summation, transposed, move_data)

copy data (involves reshape)

Arguments

TypeIntentOptionalAttributesName
type(dbcsr_tas_type), intent(inout) :: matrix_in
type(dbcsr_tas_type), intent(inout) :: matrix_out
logical, intent(in), optional :: summation

whether matrix_out = matrix_out + matrix_in

logical, intent(in), optional :: transposed
logical, intent(in), optional :: move_data

memory optimization: move data to matrix_out such that matrix_in is empty on return

public subroutine dbcsr_tas_replicate(matrix_in, info, matrix_out, nodata, move_data)

Replicate matrix_in such that each submatrix of matrix_out is an exact copy of matrix_in

Arguments

TypeIntentOptionalAttributesName
type(dbcsr_type), intent(inout) :: matrix_in
type(dbcsr_tas_split_info), intent(in) :: info
type(dbcsr_tas_type), intent(out) :: matrix_out
logical, intent(in), optional :: nodata

Don't copy data but create matrix_out

logical, intent(in), optional :: move_data

memory optimization: move data to matrix_out such that matrix_in is empty on return

public subroutine dbcsr_tas_merge(matrix_out, matrix_in, summation, move_data)

Merge submatrices of matrix_in to matrix_out by sum

Arguments

TypeIntentOptionalAttributesName
type(dbcsr_type), intent(inout) :: matrix_out
type(dbcsr_tas_type), intent(inout) :: matrix_in
logical, intent(in), optional :: summation
logical, intent(in), optional :: move_data

memory optimization: move data to matrix_out such that matrix_in is empty on return

private subroutine block_buffer_get_index(buffer, index)

get all indices from buffer

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(in) :: buffer
integer(kind=int_8), intent(out), ALLOCATABLE, DIMENSION(:, :):: index

private subroutine block_buffer_create(buffer, nblock, ndata, data_type)

Create block buffer for MPI communication.

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(out) :: buffer

block buffer

integer, intent(in) :: nblock

number of blocks total number of block entries

integer, intent(in) :: ndata

number of blocks total number of block entries

integer, intent(in) :: data_type

number of blocks total number of block entries

private subroutine block_buffer_destroy(buffer)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer

private subroutine block_buffer_add_area_block(buffer, index, block, transposed)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer(kind=int_8), intent(in), DIMENSION(2):: index
type(dbcsr_data_obj), intent(in) :: block
logical, intent(in), optional :: transposed

private subroutine block_buffer_get_next_area_block(buffer, ndata, index, block, advance_iter)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer, intent(out) :: ndata
integer(kind=int_8), intent(out), DIMENSION(2):: index
type(dbcsr_data_obj), intent(inout), optional :: block
logical, intent(in), optional :: advance_iter

private subroutine block_buffer_add_block_r_dp(buffer, index, block, transposed)

insert a block into block buffer (at current iterator position)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer(kind=int_8), intent(in), DIMENSION(2):: index

index of block

real(kind=real_8), intent(in), DIMENSION(:, :):: block

block

logical, intent(in), optional :: transposed

private subroutine block_buffer_add_block_r_sp(buffer, index, block, transposed)

insert a block into block buffer (at current iterator position)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer(kind=int_8), intent(in), DIMENSION(2):: index

index of block

real(kind=real_4), intent(in), DIMENSION(:, :):: block

block

logical, intent(in), optional :: transposed

private subroutine block_buffer_add_block_c_dp(buffer, index, block, transposed)

insert a block into block buffer (at current iterator position)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer(kind=int_8), intent(in), DIMENSION(2):: index

index of block

complex(kind=real_8), intent(in), DIMENSION(:, :):: block

block

logical, intent(in), optional :: transposed

private subroutine block_buffer_add_block_c_sp(buffer, index, block, transposed)

insert a block into block buffer (at current iterator position)

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer(kind=int_8), intent(in), DIMENSION(2):: index

index of block

complex(kind=real_4), intent(in), DIMENSION(:, :):: block

block

logical, intent(in), optional :: transposed

private subroutine block_buffer_get_next_block_r_dp(buffer, ndata, index, block, advance_iter)

get next block from buffer. Iterator is advanced only if block is retrieved or advance_iter.

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer, intent(out) :: ndata
integer(kind=int_8), intent(out), DIMENSION(2):: index
real(kind=real_8), intent(out), optional DIMENSION(:, :):: block
logical, intent(in), optional :: advance_iter

private subroutine block_buffer_get_next_block_r_sp(buffer, ndata, index, block, advance_iter)

get next block from buffer. Iterator is advanced only if block is retrieved or advance_iter.

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer, intent(out) :: ndata
integer(kind=int_8), intent(out), DIMENSION(2):: index
real(kind=real_4), intent(out), optional DIMENSION(:, :):: block
logical, intent(in), optional :: advance_iter

private subroutine block_buffer_get_next_block_c_dp(buffer, ndata, index, block, advance_iter)

get next block from buffer. Iterator is advanced only if block is retrieved or advance_iter.

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer, intent(out) :: ndata
integer(kind=int_8), intent(out), DIMENSION(2):: index
complex(kind=real_8), intent(out), optional DIMENSION(:, :):: block
logical, intent(in), optional :: advance_iter

private subroutine block_buffer_get_next_block_c_sp(buffer, ndata, index, block, advance_iter)

get next block from buffer. Iterator is advanced only if block is retrieved or advance_iter.

Arguments

TypeIntentOptionalAttributesName
type(block_buffer_type), intent(inout) :: buffer
integer, intent(out) :: ndata
integer(kind=int_8), intent(out), DIMENSION(2):: index
complex(kind=real_4), intent(out), optional DIMENSION(:, :):: block
logical, intent(in), optional :: advance_iter

private subroutine dbcsr_tas_communicate_buffer(mp_comm, buffer_recv, buffer_send, req_array)

communicate buffer

Arguments

TypeIntentOptionalAttributesName
integer, intent(in) :: mp_comm
type(block_buffer_type), intent(inout), DIMENSION(0:):: buffer_recv
type(block_buffer_type), intent(inout), DIMENSION(0:):: buffer_send
integer, intent(out), DIMENSION(:, :):: req_array