z_matmul Subroutine

subroutine z_matmul(A, B, C, alfa, beta)

Arguments

Type IntentOptional Attributes Name
complex(kind=8), intent(inout), dimension(:,:) :: A
complex(kind=8), intent(inout), dimension(:,:) :: B
complex(kind=8), intent(inout), dimension(:,:) :: C
complex(kind=8), optional :: alfa
complex(kind=8), optional :: beta

Calls

proc~~z_matmul~~CallsGraph proc~z_matmul z_matmul zgemm zgemm proc~z_matmul->zgemm

Source Code

subroutine z_matmul(A,B,C,alfa,beta)
  complex(8),dimension(:,:),intent(inout) :: A ![N,K]
  complex(8),dimension(:,:),intent(inout) :: B ![K,M]
  complex(8),dimension(:,:),intent(inout) :: C ![N,M]
  complex(8),optional                     :: alfa,beta
  complex(8)                              :: alfa_,beta_
  integer                                 :: N,K,M
  !
  ! C = alfa*A*B + beta*C
  !
  alfa_     = dcmplx(1d0,0d0) ;if(present(alfa))alfa_=alfa
  beta_     = dcmplx(0d0,0d0) ;if(present(beta))beta_=beta
  !
  !
  N = size(A,1)
  K = size(A,2) !==size(B,1)
  M = size(B,2)
  if(any(shape(B)/=[K,M]))stop "z_matmul error: B has illegal shape"
  if(any(shape(C)/=[N,M]))stop "z_matmul error: C has illegal shape"
  !
  call ZGEMM('N', 'N', N, M, K, alfa_, A, N, B, K, beta_, C, N)
  !
  return
end subroutine z_matmul