|
PROGRAM TEST
|
IMPLICIT NONE
|
C
|
INTEGER LDA, LDAF, LDB, LDIWRK, LDWORK, LDX
|
INTEGER N, NRHS
|
PARAMETER (N = 3)
|
PARAMETER (LDA = N)
|
PARAMETER (LDAF = N)
|
PARAMETER (LDB = N)
|
PARAMETER (LDIWRK = N)
|
PARAMETER (LDWORK = 3 * N)
|
PARAMETER (LDX = N)
|
PARAMETER (NRHS = 1)
|
C
|
DOUBLE PRECISION A(LDA,N), AF(LDAF,N), B(LDB,NRHS)
|
DOUBLE PRECISION B1(LDB,NRHS), BERR(NRHS), EPSLON
|
DOUBLE PRECISION FERR(NRHS), WORK(LDWORK), X(LDX,NRHS)
|
INTEGER ICOL, INFO, IPIVOT(N), IROW
|
INTEGER IWORK(LDIWRK)
|
C
|
EXTERNAL DCOPY, DGEMV, DGERFS, DGETRF, DGETRS
|
INTRINSIC ABS, DBLE
|
C
|
EPSLON = ((((2.0D0 / 3.0D0) + 1.6D1) - 1.6D1) -
|
$ (2.0D0 / 3.0D0))
|
C
|
C Initialize the array A to store the matrix A shown below.
|
C Each of the diagonal elements is equal to the row number
|
C plus epsilon.
|
C
|
C 1+e 1 1
|
C A = 2 2+e 2
|
C 3 3 3+e
|
C
|
DO 110, ICOL = 1, N
|
DO 100, IROW = 1, N
|
A(IROW,ICOL) = DBLE(IROW)
|
100 CONTINUE
|
A(ICOL,ICOL) = A(ICOL,ICOL) + EPSLON
|
110 CONTINUE
|
CALL DCOPY (LDA*N, A, 1, AF, 1)
|
C
|
C Print the initial value of A.
|
C
|
PRINT 1000
|
DO 120, IROW = 1, N
|
PRINT 1010, (A(IROW,ICOL), ICOL = 1, N)
|
120 CONTINUE
|
C
|
C Initialize the array B to store the right hand side
|
C vector b shown below. Each element is equal to N plus
|
C a small epsilon. Print the contents of the array after
|
C initialization.
|
C
|
C N+e
|
C b = N+e
|
C N+e
|
C
|
DO 130, IROW = 1, N
|
B(IROW,1) = DBLE(N) + EPSLON
|
130 CONTINUE
|
CALL DCOPY (N, B, 1, X, 1)
|
PRINT 1020
|
PRINT 1030, B
|
C
|
C LU factor A.
|
C
|
CALL DGETRF (N, N, AF, LDAF, IPIVOT, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1040, INFO
|
STOP 1
|
END IF
|
C
|
C Use the LU factorization to solve the system.
|
C
|
CALL DGETRS ('NO TRANSPOSE A', N, NRHS, AF, LDAF, IPIVOT,
|
$ X, LDX, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1050, INFO
|
STOP 2
|
END IF
|
C
|
C Print the solution and Ax.
|
C
|
PRINT 1060
|
PRINT 1070, X
|
CALL DGEMV ('N', N, N, 1.0D0, A, LDA, X, 1, 0.0D0, B1, 1)
|
PRINT 1080
|
PRINT 1030, B1
|
C
|
C Refine the solution, then print the refined solution and Ax
|
C with the refined x.
|
C
|
CALL DGERFS ('NO TRANSPOSE A', N, NRHS, A, LDA, AF, LDAF,
|
$ IPIVOT, B, LDB, X, LDX, FERR, BERR, WORK,
|
$ IWORK, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1090, INFO
|
STOP 2
|
END IF
|
PRINT 1100
|
PRINT 1070, X
|
CALL DGEMV ('N', N, N, 1.0D0, A, LDA, X, 1, 0.0D0, B1, 1)
|
PRINT 1110
|
PRINT 1030, B1
|
C
|
1000 FORMAT (1X, 'A:')
|
1010 FORMAT (3(3X, F22.17))
|
1020 FORMAT (/1X, 'b:')
|
1030 FORMAT (1X, F22.17)
|
1040 FORMAT (1X, 'Error factoring A in DGETRF, INFO = ', I5)
|
1050 FORMAT (1X, 'Error solving Ax=b in DGETRF, INFO = ', I5)
|
1060 FORMAT (/1X, 'Initial solution for Ax=b:')
|
1070 FORMAT (1X, E16.6)
|
1080 FORMAT (/1X, 'Ax with initial x:')
|
1090 FORMAT (1X, 'Error improving solution in DGERFS,
|
$ INFO = ', I5)
|
1100 FORMAT (/1X, 'Improved solution for Ax=b:')
|
1110 FORMAT (/1X, 'Ax with improved x:')
|
C
|
END
|
|