LAPACK 3.11.0
LAPACK: Linear Algebra PACKage
Loading...
Searching...
No Matches

◆ zchkhe_aa_2stage()

subroutine zchkhe_aa_2stage ( logical, dimension( * )  DOTYPE,
integer  NN,
integer, dimension( * )  NVAL,
integer  NNB,
integer, dimension( * )  NBVAL,
integer  NNS,
integer, dimension( * )  NSVAL,
double precision  THRESH,
logical  TSTERR,
integer  NMAX,
complex*16, dimension( * )  A,
complex*16, dimension( * )  AFAC,
complex*16, dimension( * )  AINV,
complex*16, dimension( * )  B,
complex*16, dimension( * )  X,
complex*16, dimension( * )  XACT,
complex*16, dimension( * )  WORK,
double precision, dimension( * )  RWORK,
integer, dimension( * )  IWORK,
integer  NOUT 
)

ZCHKHE_AA_2STAGE

Purpose:
 ZCHKSY_AA_2STAGE tests ZHETRF_AA_2STAGE, -TRS_AA_2STAGE.
Parameters
[in]DOTYPE
          DOTYPE is LOGICAL array, dimension (NTYPES)
          The matrix types to be used for testing.  Matrices of type j
          (for 1 <= j <= NTYPES) are used for testing if DOTYPE(j) =
          .TRUE.; if DOTYPE(j) = .FALSE., then type j is not used.
[in]NN
          NN is INTEGER
          The number of values of N contained in the vector NVAL.
[in]NVAL
          NVAL is INTEGER array, dimension (NN)
          The values of the matrix dimension N.
[in]NNB
          NNB is INTEGER
          The number of values of NB contained in the vector NBVAL.
[in]NBVAL
          NBVAL is INTEGER array, dimension (NNB)
          The values of the blocksize NB.
[in]NNS
          NNS is INTEGER
          The number of values of NRHS contained in the vector NSVAL.
[in]NSVAL
          NSVAL is INTEGER array, dimension (NNS)
          The values of the number of right hand sides NRHS.
[in]THRESH
          THRESH is DOUBLE PRECISION
          The threshold value for the test ratios.  A result is
          included in the output file if RESULT >= THRESH.  To have
          every test ratio printed, use THRESH = 0.
[in]TSTERR
          TSTERR is LOGICAL
          Flag that indicates whether error exits are to be tested.
[in]NMAX
          NMAX is INTEGER
          The maximum value permitted for N, used in dimensioning the
          work arrays.
[out]A
          A is COMPLEX*16 array, dimension (NMAX*NMAX)
[out]AFAC
          AFAC is COMPLEX*16 array, dimension (NMAX*NMAX)
[out]AINV
          AINV is COMPLEX*16 array, dimension (NMAX*NMAX)
[out]B
          B is COMPLEX*16 array, dimension (NMAX*NSMAX)
          where NSMAX is the largest entry in NSVAL.
[out]X
          X is COMPLEX*16 array, dimension (NMAX*NSMAX)
[out]XACT
          XACT is COMPLEX*16 array, dimension (NMAX*NSMAX)
[out]WORK
          WORK is COMPLEX*16 array, dimension (NMAX*max(3,NSMAX))
[out]RWORK
          RWORK is DOUBLE PRECISION array, dimension (max(NMAX,2*NSMAX))
[out]IWORK
          IWORK is INTEGER array, dimension (2*NMAX)
[in]NOUT
          NOUT is INTEGER
          The unit number for output.
Author
Univ. of Tennessee
Univ. of California Berkeley
Univ. of Colorado Denver
NAG Ltd.

Definition at line 169 of file zchkhe_aa_2stage.f.

172*
173* -- LAPACK test routine --
174* -- LAPACK is a software package provided by Univ. of Tennessee, --
175* -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
176*
177 IMPLICIT NONE
178*
179* .. Scalar Arguments ..
180 LOGICAL TSTERR
181 INTEGER NN, NNB, NNS, NMAX, NOUT
182 DOUBLE PRECISION THRESH
183* ..
184* .. Array Arguments ..
185 LOGICAL DOTYPE( * )
186 INTEGER IWORK( * ), NBVAL( * ), NSVAL( * ), NVAL( * )
187 COMPLEX*16 A( * ), AFAC( * ), AINV( * ), B( * ),
188 $ WORK( * ), X( * ), XACT( * )
189 DOUBLE PRECISION RWORK( * )
190* ..
191*
192* =====================================================================
193*
194* .. Parameters ..
195 DOUBLE PRECISION ZERO
196 parameter( zero = 0.0d+0 )
197 COMPLEX*16 CZERO
198 parameter( czero = ( 0.0d+0, 0.0d+0 ) )
199 INTEGER NTYPES
200 parameter( ntypes = 10 )
201 INTEGER NTESTS
202 parameter( ntests = 9 )
203* ..
204* .. Local Scalars ..
205 LOGICAL ZEROT
206 CHARACTER DIST, TYPE, UPLO, XTYPE
207 CHARACTER*3 PATH, MATPATH
208 INTEGER I, I1, I2, IMAT, IN, INB, INFO, IOFF, IRHS,
209 $ IUPLO, IZERO, J, K, KL, KU, LDA, LWORK, MODE,
210 $ N, NB, NERRS, NFAIL, NIMAT, NRHS, NRUN, NT
211 DOUBLE PRECISION ANORM, CNDNUM
212* ..
213* .. Local Arrays ..
214 CHARACTER UPLOS( 2 )
215 INTEGER ISEED( 4 ), ISEEDY( 4 )
216 DOUBLE PRECISION RESULT( NTESTS )
217* ..
218* .. External Subroutines ..
219 EXTERNAL alaerh, alahd, alasum, zerrhe, zlacpy,
222 $ xlaenv
223* ..
224* .. Intrinsic Functions ..
225 INTRINSIC max, min
226* ..
227* .. Scalars in Common ..
228 LOGICAL LERR, OK
229 CHARACTER*32 SRNAMT
230 INTEGER INFOT, NUNIT
231* ..
232* .. Common blocks ..
233 COMMON / infoc / infot, nunit, ok, lerr
234 COMMON / srnamc / srnamt
235* ..
236* .. Data statements ..
237 DATA iseedy / 1988, 1989, 1990, 1991 /
238 DATA uplos / 'U', 'L' /
239* ..
240* .. Executable Statements ..
241*
242* Initialize constants and the random number seed.
243*
244* Test path
245*
246 path( 1: 1 ) = 'Zomplex precision'
247 path( 2: 3 ) = 'H2'
248*
249* Path to generate matrices
250*
251 matpath( 1: 1 ) = 'Zomplex precision'
252 matpath( 2: 3 ) = 'HE'
253 nrun = 0
254 nfail = 0
255 nerrs = 0
256 DO 10 i = 1, 4
257 iseed( i ) = iseedy( i )
258 10 CONTINUE
259*
260* Test the error exits
261*
262 IF( tsterr )
263 $ CALL zerrhe( path, nout )
264 infot = 0
265*
266* Set the minimum block size for which the block routine should
267* be used, which will be later returned by ILAENV
268*
269 CALL xlaenv( 2, 2 )
270*
271* Do for each value of N in NVAL
272*
273 DO 180 in = 1, nn
274 n = nval( in )
275 IF( n .GT. nmax ) THEN
276 nfail = nfail + 1
277 WRITE(nout, 9995) 'M ', n, nmax
278 GO TO 180
279 END IF
280 lda = max( n, 1 )
281 xtype = 'N'
282 nimat = ntypes
283 IF( n.LE.0 )
284 $ nimat = 1
285*
286 izero = 0
287*
288* Do for each value of matrix type IMAT
289*
290 DO 170 imat = 1, nimat
291*
292* Do the tests only if DOTYPE( IMAT ) is true.
293*
294 IF( .NOT.dotype( imat ) )
295 $ GO TO 170
296*
297* Skip types 3, 4, 5, or 6 if the matrix size is too small.
298*
299 zerot = imat.GE.3 .AND. imat.LE.6
300 IF( zerot .AND. n.LT.imat-2 )
301 $ GO TO 170
302*
303* Do first for UPLO = 'U', then for UPLO = 'L'
304*
305 DO 160 iuplo = 1, 2
306 uplo = uplos( iuplo )
307*
308* Begin generate the test matrix A.
309*
310*
311* Set up parameters with ZLATB4 for the matrix generator
312* based on the type of matrix to be generated.
313*
314 CALL zlatb4( matpath, imat, n, n, TYPE, KL, KU,
315 $ ANORM, MODE, CNDNUM, DIST )
316*
317* Generate a matrix with ZLATMS.
318*
319 srnamt = 'ZLATMS'
320 CALL zlatms( n, n, dist, iseed, TYPE, RWORK, MODE,
321 $ CNDNUM, ANORM, KL, KU, UPLO, A, LDA, WORK,
322 $ INFO )
323*
324* Check error code from ZLATMS and handle error.
325*
326 IF( info.NE.0 ) THEN
327 CALL alaerh( path, 'ZLATMS', info, 0, uplo, n, n, -1,
328 $ -1, -1, imat, nfail, nerrs, nout )
329*
330* Skip all tests for this generated matrix
331*
332 GO TO 160
333 END IF
334*
335* For matrix types 3-6, zero one or more rows and
336* columns of the matrix to test that INFO is returned
337* correctly.
338*
339 IF( zerot ) THEN
340 IF( imat.EQ.3 ) THEN
341 izero = 1
342 ELSE IF( imat.EQ.4 ) THEN
343 izero = n
344 ELSE
345 izero = n / 2 + 1
346 END IF
347*
348 IF( imat.LT.6 ) THEN
349*
350* Set row and column IZERO to zero.
351*
352 IF( iuplo.EQ.1 ) THEN
353 ioff = ( izero-1 )*lda
354 DO 20 i = 1, izero - 1
355 a( ioff+i ) = czero
356 20 CONTINUE
357 ioff = ioff + izero
358 DO 30 i = izero, n
359 a( ioff ) = czero
360 ioff = ioff + lda
361 30 CONTINUE
362 ELSE
363 ioff = izero
364 DO 40 i = 1, izero - 1
365 a( ioff ) = czero
366 ioff = ioff + lda
367 40 CONTINUE
368 ioff = ioff - izero
369 DO 50 i = izero, n
370 a( ioff+i ) = czero
371 50 CONTINUE
372 END IF
373 ELSE
374 IF( iuplo.EQ.1 ) THEN
375*
376* Set the first IZERO rows and columns to zero.
377*
378 ioff = 0
379 DO 70 j = 1, n
380 i2 = min( j, izero )
381 DO 60 i = 1, i2
382 a( ioff+i ) = czero
383 60 CONTINUE
384 ioff = ioff + lda
385 70 CONTINUE
386 izero = 1
387 ELSE
388*
389* Set the last IZERO rows and columns to zero.
390*
391 ioff = 0
392 DO 90 j = 1, n
393 i1 = max( j, izero )
394 DO 80 i = i1, n
395 a( ioff+i ) = czero
396 80 CONTINUE
397 ioff = ioff + lda
398 90 CONTINUE
399 END IF
400 END IF
401 ELSE
402 izero = 0
403 END IF
404*
405* End generate test matrix A.
406*
407*
408* Set the imaginary part of the diagonals.
409*
410 CALL zlaipd( n, a, lda+1, 0 )
411*
412* Do for each value of NB in NBVAL
413*
414 DO 150 inb = 1, nnb
415*
416* Set the optimal blocksize, which will be later
417* returned by ILAENV.
418*
419 nb = nbval( inb )
420 CALL xlaenv( 1, nb )
421*
422* Copy the test matrix A into matrix AFAC which
423* will be factorized in place. This is needed to
424* preserve the test matrix A for subsequent tests.
425*
426 CALL zlacpy( uplo, n, n, a, lda, afac, lda )
427*
428* Compute the L*D*L**T or U*D*U**T factorization of the
429* matrix. IWORK stores details of the interchanges and
430* the block structure of D. AINV is a work array for
431* block factorization, LWORK is the length of AINV.
432*
433 srnamt = 'ZHETRF_AA_2STAGE'
434 lwork = min(n*nb, 3*nmax*nmax)
435 CALL zhetrf_aa_2stage( uplo, n, afac, lda,
436 $ ainv, (3*nb+1)*n,
437 $ iwork, iwork( 1+n ),
438 $ work, lwork,
439 $ info )
440*
441* Adjust the expected value of INFO to account for
442* pivoting.
443*
444 IF( izero.GT.0 ) THEN
445 j = 1
446 k = izero
447 100 CONTINUE
448 IF( j.EQ.k ) THEN
449 k = iwork( j )
450 ELSE IF( iwork( j ).EQ.k ) THEN
451 k = j
452 END IF
453 IF( j.LT.k ) THEN
454 j = j + 1
455 GO TO 100
456 END IF
457 ELSE
458 k = 0
459 END IF
460*
461* Check error code from CHETRF and handle error.
462*
463 IF( info.NE.k ) THEN
464 CALL alaerh( path, 'ZHETRF_AA_2STAGE', info, k,
465 $ uplo, n, n, -1, -1, nb, imat, nfail,
466 $ nerrs, nout )
467 END IF
468*
469*+ TEST 1
470* Reconstruct matrix from factors and compute residual.
471*
472* NEED TO CREATE ZHET01_AA_2STAGE
473* CALL ZHET01_AA( UPLO, N, A, LDA, AFAC, LDA, IWORK,
474* $ AINV, LDA, RWORK, RESULT( 1 ) )
475* NT = 1
476 nt = 0
477*
478*
479* Print information about the tests that did not pass
480* the threshold.
481*
482 DO 110 k = 1, nt
483 IF( result( k ).GE.thresh ) THEN
484 IF( nfail.EQ.0 .AND. nerrs.EQ.0 )
485 $ CALL alahd( nout, path )
486 WRITE( nout, fmt = 9999 )uplo, n, nb, imat, k,
487 $ result( k )
488 nfail = nfail + 1
489 END IF
490 110 CONTINUE
491 nrun = nrun + nt
492*
493* Skip solver test if INFO is not 0.
494*
495 IF( info.NE.0 ) THEN
496 GO TO 140
497 END IF
498*
499* Do for each value of NRHS in NSVAL.
500*
501 DO 130 irhs = 1, nns
502 nrhs = nsval( irhs )
503*
504*+ TEST 2 (Using TRS)
505* Solve and compute residual for A * X = B.
506*
507* Choose a set of NRHS random solution vectors
508* stored in XACT and set up the right hand side B
509*
510 srnamt = 'ZLARHS'
511 CALL zlarhs( matpath, xtype, uplo, ' ', n, n,
512 $ kl, ku, nrhs, a, lda, xact, lda,
513 $ b, lda, iseed, info )
514 CALL zlacpy( 'Full', n, nrhs, b, lda, x, lda )
515*
516 srnamt = 'ZHETRS_AA_2STAGE'
517 lwork = max( 1, 3*n-2 )
518 CALL zhetrs_aa_2stage( uplo, n, nrhs, afac, lda,
519 $ ainv, (3*nb+1)*n, iwork, iwork( 1+n ),
520 $ x, lda, info )
521*
522* Check error code from ZHETRS and handle error.
523*
524 IF( info.NE.0 ) THEN
525 IF( izero.EQ.0 ) THEN
526 CALL alaerh( path, 'ZHETRS_AA_2STAGE',
527 $ info, 0, uplo, n, n, -1, -1,
528 $ nrhs, imat, nfail, nerrs, nout )
529 END IF
530 ELSE
531*
532 CALL zlacpy( 'Full', n, nrhs, b, lda, work, lda
533 $ )
534*
535* Compute the residual for the solution
536*
537 CALL zpot02( uplo, n, nrhs, a, lda, x, lda,
538 $ work, lda, rwork, result( 2 ) )
539*
540* Print information about the tests that did not pass
541* the threshold.
542*
543 DO 120 k = 2, 2
544 IF( result( k ).GE.thresh ) THEN
545 IF( nfail.EQ.0 .AND. nerrs.EQ.0 )
546 $ CALL alahd( nout, path )
547 WRITE( nout, fmt = 9998 )uplo, n, nrhs,
548 $ imat, k, result( k )
549 nfail = nfail + 1
550 END IF
551 120 CONTINUE
552 END IF
553 nrun = nrun + 1
554*
555* End do for each value of NRHS in NSVAL.
556*
557 130 CONTINUE
558 140 CONTINUE
559 150 CONTINUE
560 160 CONTINUE
561 170 CONTINUE
562 180 CONTINUE
563*
564* Print a summary of the results.
565*
566 CALL alasum( path, nout, nfail, nrun, nerrs )
567*
568 9999 FORMAT( ' UPLO = ''', a1, ''', N =', i5, ', NB =', i4, ', type ',
569 $ i2, ', test ', i2, ', ratio =', g12.5 )
570 9998 FORMAT( ' UPLO = ''', a1, ''', N =', i5, ', NRHS=', i3, ', type ',
571 $ i2, ', test(', i2, ') =', g12.5 )
572 9995 FORMAT( ' Invalid input value: ', a4, '=', i6, '; must be <=',
573 $ i6 )
574 RETURN
575*
576* End of ZCHKHE_AA_2STAGE
577*
subroutine alasum(TYPE, NOUT, NFAIL, NRUN, NERRS)
ALASUM
Definition: alasum.f:73
subroutine xlaenv(ISPEC, NVALUE)
XLAENV
Definition: xlaenv.f:81
subroutine alahd(IOUNIT, PATH)
ALAHD
Definition: alahd.f:107
subroutine alaerh(PATH, SUBNAM, INFO, INFOE, OPTS, M, N, KL, KU, N5, IMAT, NFAIL, NERRS, NOUT)
ALAERH
Definition: alaerh.f:147
subroutine zlarhs(PATH, XTYPE, UPLO, TRANS, M, N, KL, KU, NRHS, A, LDA, X, LDX, B, LDB, ISEED, INFO)
ZLARHS
Definition: zlarhs.f:208
subroutine zerrhe(PATH, NUNIT)
ZERRHE
Definition: zerrhe.f:55
subroutine zlaipd(N, A, INDA, VINDA)
ZLAIPD
Definition: zlaipd.f:83
subroutine zpot02(UPLO, N, NRHS, A, LDA, X, LDX, B, LDB, RWORK, RESID)
ZPOT02
Definition: zpot02.f:127
subroutine zlatb4(PATH, IMAT, M, N, TYPE, KL, KU, ANORM, MODE, CNDNUM, DIST)
ZLATB4
Definition: zlatb4.f:121
subroutine zlatms(M, N, DIST, ISEED, SYM, D, MODE, COND, DMAX, KL, KU, PACK, A, LDA, WORK, INFO)
ZLATMS
Definition: zlatms.f:332
subroutine zlacpy(UPLO, M, N, A, LDA, B, LDB)
ZLACPY copies all or part of one two-dimensional array to another.
Definition: zlacpy.f:103
subroutine zhetrs_aa_2stage(UPLO, N, NRHS, A, LDA, TB, LTB, IPIV, IPIV2, B, LDB, INFO)
ZHETRS_AA_2STAGE
subroutine zhetrf_aa_2stage(UPLO, N, A, LDA, TB, LTB, IPIV, IPIV2, WORK, LWORK, INFO)
ZHETRF_AA_2STAGE
Here is the call graph for this function:
Here is the caller graph for this function: