46488e8cbc
Added an example code to test CG on a 4D HYPRE_SSTRUCT complex problem. Added regression tests for bigint, maxdim, and complex. Added a test to make sure double types are not added to the source. See [Issue995] in the tracker for more details.
308 lines
10 KiB
C
308 lines
10 KiB
C
/*BHEADER**********************************************************************
|
|
* Copyright (c) 2008, Lawrence Livermore National Security, LLC.
|
|
* Produced at the Lawrence Livermore National Laboratory.
|
|
* This file is part of HYPRE. See file COPYRIGHT for details.
|
|
*
|
|
* HYPRE is free software; you can redistribute it and/or modify it under the
|
|
* terms of the GNU Lesser General Public License (as published by the Free
|
|
* Software Foundation) version 2.1 dated February 1999.
|
|
*
|
|
* $Revision$
|
|
***********************************************************************EHEADER*/
|
|
|
|
#include "_hypre_parcsr_ls.h"
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGCreate
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGCreate( MPI_Comm comm, HYPRE_Solver *solver )
|
|
{
|
|
hypre_PCGFunctions * pcg_functions;
|
|
|
|
if (!solver)
|
|
{
|
|
hypre_error_in_arg(2);
|
|
return hypre_error_flag;
|
|
}
|
|
pcg_functions =
|
|
hypre_PCGFunctionsCreate(
|
|
hypre_CAlloc, hypre_ParKrylovFree, hypre_ParKrylovCommInfo,
|
|
hypre_ParKrylovCreateVector,
|
|
hypre_ParKrylovDestroyVector, hypre_ParKrylovMatvecCreate,
|
|
hypre_ParKrylovMatvec, hypre_ParKrylovMatvecDestroy,
|
|
hypre_ParKrylovInnerProd, hypre_ParKrylovCopyVector,
|
|
hypre_ParKrylovClearVector,
|
|
hypre_ParKrylovScaleVector, hypre_ParKrylovAxpy,
|
|
hypre_ParKrylovIdentitySetup, hypre_ParKrylovIdentity );
|
|
*solver = ( (HYPRE_Solver) hypre_PCGCreate( pcg_functions ) );
|
|
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGDestroy
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGDestroy( HYPRE_Solver solver )
|
|
{
|
|
return( hypre_PCGDestroy( (void *) solver ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetup
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetup( HYPRE_Solver solver,
|
|
HYPRE_ParCSRMatrix A,
|
|
HYPRE_ParVector b,
|
|
HYPRE_ParVector x )
|
|
{
|
|
return( HYPRE_PCGSetup( solver,
|
|
(HYPRE_Matrix) A,
|
|
(HYPRE_Vector) b,
|
|
(HYPRE_Vector) x ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSolve
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSolve( HYPRE_Solver solver,
|
|
HYPRE_ParCSRMatrix A,
|
|
HYPRE_ParVector b,
|
|
HYPRE_ParVector x )
|
|
{
|
|
return( HYPRE_PCGSolve( solver,
|
|
(HYPRE_Matrix) A,
|
|
(HYPRE_Vector) b,
|
|
(HYPRE_Vector) x ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetTol
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetTol( HYPRE_Solver solver,
|
|
HYPRE_Real tol )
|
|
{
|
|
return( HYPRE_PCGSetTol( solver, tol ) );
|
|
}
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetAbsoluteTol
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetAbsoluteTol( HYPRE_Solver solver,
|
|
HYPRE_Real a_tol )
|
|
{
|
|
return( HYPRE_PCGSetAbsoluteTol( solver, a_tol ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetMaxIter
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetMaxIter( HYPRE_Solver solver,
|
|
HYPRE_Int max_iter )
|
|
{
|
|
return( HYPRE_PCGSetMaxIter( solver, max_iter ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetStopCrit
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetStopCrit( HYPRE_Solver solver,
|
|
HYPRE_Int stop_crit )
|
|
{
|
|
return( HYPRE_PCGSetStopCrit( solver, stop_crit ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetTwoNorm
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetTwoNorm( HYPRE_Solver solver,
|
|
HYPRE_Int two_norm )
|
|
{
|
|
return( HYPRE_PCGSetTwoNorm( solver, two_norm ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetRelChange
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetRelChange( HYPRE_Solver solver,
|
|
HYPRE_Int rel_change )
|
|
{
|
|
return( HYPRE_PCGSetRelChange( solver, rel_change ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetPrecond
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetPrecond( HYPRE_Solver solver,
|
|
HYPRE_PtrToParSolverFcn precond,
|
|
HYPRE_PtrToParSolverFcn precond_setup,
|
|
HYPRE_Solver precond_solver )
|
|
{
|
|
return( HYPRE_PCGSetPrecond( solver,
|
|
(HYPRE_PtrToSolverFcn) precond,
|
|
(HYPRE_PtrToSolverFcn) precond_setup,
|
|
precond_solver ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGGetPrecond
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGGetPrecond( HYPRE_Solver solver,
|
|
HYPRE_Solver *precond_data_ptr )
|
|
{
|
|
return( HYPRE_PCGGetPrecond( solver, precond_data_ptr ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetPrintLevel
|
|
* an obsolete function; use HYPRE_PCG* functions instead
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetPrintLevel( HYPRE_Solver solver,
|
|
HYPRE_Int level )
|
|
{
|
|
return( HYPRE_PCGSetPrintLevel( solver, level ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGSetLogging
|
|
* an obsolete function; use HYPRE_PCG* functions instead
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGSetLogging( HYPRE_Solver solver,
|
|
HYPRE_Int level )
|
|
{
|
|
return( HYPRE_PCGSetLogging( solver, level ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGGetNumIterations
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGGetNumIterations( HYPRE_Solver solver,
|
|
HYPRE_Int *num_iterations )
|
|
{
|
|
return( HYPRE_PCGGetNumIterations( solver, num_iterations ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRPCGGetFinalRelativeResidualNorm
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRPCGGetFinalRelativeResidualNorm( HYPRE_Solver solver,
|
|
HYPRE_Real *norm )
|
|
{
|
|
return( HYPRE_PCGGetFinalRelativeResidualNorm( solver, norm ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRDiagScaleSetup
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRDiagScaleSetup( HYPRE_Solver solver,
|
|
HYPRE_ParCSRMatrix A,
|
|
HYPRE_ParVector y,
|
|
HYPRE_ParVector x )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRDiagScale
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRDiagScale( HYPRE_Solver solver,
|
|
HYPRE_ParCSRMatrix HA,
|
|
HYPRE_ParVector Hy,
|
|
HYPRE_ParVector Hx )
|
|
{
|
|
hypre_ParCSRMatrix *A = (hypre_ParCSRMatrix *) HA;
|
|
hypre_ParVector *y = (hypre_ParVector *) Hy;
|
|
hypre_ParVector *x = (hypre_ParVector *) Hx;
|
|
HYPRE_Real *x_data = hypre_VectorData(hypre_ParVectorLocalVector(x));
|
|
HYPRE_Real *y_data = hypre_VectorData(hypre_ParVectorLocalVector(y));
|
|
HYPRE_Real *A_data = hypre_CSRMatrixData(hypre_ParCSRMatrixDiag(A));
|
|
HYPRE_Int *A_i = hypre_CSRMatrixI(hypre_ParCSRMatrixDiag(A));
|
|
HYPRE_Int local_size = hypre_VectorSize(hypre_ParVectorLocalVector(x));
|
|
HYPRE_Int i, ierr = 0;
|
|
|
|
for (i=0; i < local_size; i++)
|
|
{
|
|
x_data[i] = y_data[i]/A_data[A_i[i]];
|
|
}
|
|
|
|
return ierr;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* HYPRE_ParCSRSymPrecondSetup
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
/*
|
|
|
|
HYPRE_Int
|
|
HYPRE_ParCSRSymPrecondSetup( HYPRE_Solver solver,
|
|
HYPRE_ParCSRMatrix A,
|
|
HYPRE_ParVector b,
|
|
HYPRE_ParVector x )
|
|
{
|
|
hypre_ParCSRMatrix *A = (hypre_ParCSRMatrix *) A;
|
|
hypre_ParVector *y = (hypre_ParVector *) b;
|
|
hypre_ParVector *x = (hypre_ParVector *) x;
|
|
|
|
HYPRE_Real *x_data = hypre_VectorData(hypre_ParVectorLocalVector(x));
|
|
HYPRE_Real *y_data = hypre_VectorData(hypre_ParVectorLocalVector(y));
|
|
HYPRE_Real *A_diag = hypre_CSRMatrixData(hypre_ParCSRMatrixDiag(A));
|
|
HYPRE_Real *A_offd = hypre_CSRMatrixData(hypre_ParCSRMatrixOffD(A));
|
|
|
|
HYPRE_Int i, ierr = 0;
|
|
hypre_ParCSRMatrix *Asym;
|
|
MPI_Comm comm;
|
|
HYPRE_Int global_num_rows;
|
|
HYPRE_Int global_num_cols;
|
|
HYPRE_Int *row_starts;
|
|
HYPRE_Int *col_starts;
|
|
HYPRE_Int num_cols_offd;
|
|
HYPRE_Int num_nonzeros_diag;
|
|
HYPRE_Int num_nonzeros_offd;
|
|
|
|
Asym = hypre_ParCSRMatrixCreate(comm, global_num_rows, global_num_cols,
|
|
row_starts, col_starts, num_cols_offd,
|
|
num_nonzeros_diag, num_nonzeros_offd);
|
|
|
|
for (i=0; i < hypre_VectorSize(hypre_ParVectorLocalVector(x)); i++)
|
|
{
|
|
x_data[i] = y_data[i]/A_data[A_i[i]];
|
|
}
|
|
|
|
return ierr;
|
|
} */
|