[通用算法] MKL函数库中ZGETRF和ZGETRI函数的使用

PROGRAM TE
IMPLICIT NONE
INTEGER :: i, j, dime                                      !Ingeter for loop and order for complex matrix
DOUBLE COMPLEX, ALLOCATABLE :: a(:,:)       !Original complex matrix
DOUBLE COMPLEX, ALLOCATABLE :: b(:,:)       !Inversed complex matrix
DOUBLE COMPLEX, ALLOCATABLE :: c(:,:)       !Multiplication of the original andinversed complex matrices

!Variables used in LU decomposition subroutine (ZGETRF)
INTEGER                                     :: nu_r        !Number ofrow in target matrix
INTEGER                                     :: nu_c        !Number ofcolumn in target matrix
INTEGER                                     :: lu_lda      !Leadingdimension of target matrix
INTEGER, ALLOCATABLE               :: lu_ipiv(:)  !Arraywith dimension of .GE. MIN(nu_r,nu_c)
INTEGER                                     :: lu_info     !Judgement(0-successful;<0-error and infor = number, then
                                                                    !'1-number'th arguement hasillegal value;>0 and info =
                                                                    !number - LU matrix(number,number) is zero)

!Variables used in inverse subroutine (ZGETRI)
INTEGER                                     :: nu_o        !Order ofLU matrix
DOUBLE COMPLEX, ALLOCATABLE :: in_work(:) !Workspacearray with dimension of .GE. MAX(1,in_lwork)
INTEGER                      :: in_lda                     !Leadingdimension of LU decomposed matrix .GE. MAX(1,nu_o)
INTEGER                      :: in_lwork                  !Size ofin_work array with value .GE. nu_o
INTEGER                      :: in_info                    !Judgement(0-successful;<0-error and infor = number, then
                                                                    !'1-number'th arguement hasillegal value;>0 and info =
                                                                    !number; then, LU matrix(number,number) is zero and inversion
                                                                    !cannot be accomplished)

dime = 3                                                       !Thisparameter is used to control the size of the complex matrix
ALLOCATE (a(dime,dime))
ALLOCATE (b(dime,dime))
ALLOCATE (c(dime,dime))

DO i = 1, dime, 1
   DO j = 1, dime, 1
      a(i,j) =CMPLX(DBLE(i), DBLE(j))
   END DO
END DO

b = a

OPEN (UNIT=3, FILE='te_in.dat', STATUS='UNKNOWN')
WRITE (UNIT=3, FMT='(A1)') 'a'
DO i = 1, dime, 1
   DO j = 1, dime, 1
      WRITE (UNIT=3,FMT=*) a(i,j)
   END DO
END DO
WRITE (UNIT=3, FMT=*)

!Initialising parameters for LU decomposition subroutine(ZGETRF)
nu_r = dime
nu_c = dime
lu_lda = dime
ALLOCATE (lu_ipiv(dime))
lu_ipiv = 0
!

!Initialising parameters for inverse subroutine (ZGETRI)
nu_o = dime
in_lwork = dime
ALLOCATE (in_work(in_lwork))
in_work = (0.0d0, 0.0d0)
in_lda = dime
!

CALL ZGETRF(nu_r,nu_c,b,lu_lda,lu_ipiv,lu_info)
CALL ZGETRI(nu_o,b,in_lda,lu_ipiv,in_work,in_lwork,in_info)

WRITE (UNIT=3, FMT='(A1)') 'b'
DO i = 1, dime, 1
   DO j = 1, dime, 1
      WRITE (UNIT=3,FMT=*) b(i,j)
   END DO
END DO
WRITE (UNIT=3, FMT=*)

c = MATMUL(a,b)

WRITE (UNIT=3, FMT='(A1)') 'c'
DO i = 1, dime, 1
   DO j = 1, dime, 1
      WRITE (UNIT=3,FMT=*) c(i,j)
   END DO
END DO
WRITE (UNIT=3, FMT=*)

DEALLOCATE (lu_ipiv)
DEALLOCATE (a)
DEALLOCATE (b)
DEALLOCATE (c)

CLOSE (UNIT=3)
STOP
END PROGRAM TE