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.
280 lines
9.0 KiB
C
280 lines
9.0 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*/
|
|
|
|
/******************************************************************************
|
|
*
|
|
* SStruct matrix-vector implementation of Krylov interface routines.
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#include "_hypre_sstruct_ls.h"
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
char *
|
|
hypre_SStructKrylovCAlloc( HYPRE_Int count,
|
|
HYPRE_Int elt_size )
|
|
{
|
|
return( hypre_CAlloc( count, elt_size ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovFree( char *ptr )
|
|
{
|
|
hypre_Free( ptr );
|
|
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
void *
|
|
hypre_SStructKrylovCreateVector( void *vvector )
|
|
{
|
|
hypre_SStructVector *vector = vvector;
|
|
hypre_SStructVector *new_vector;
|
|
HYPRE_Int object_type;
|
|
|
|
HYPRE_Int nparts= hypre_SStructVectorNParts(vector);
|
|
hypre_SStructPVector *pvector;
|
|
hypre_StructVector *svector;
|
|
hypre_SStructPVector *new_pvector;
|
|
hypre_StructVector *new_svector;
|
|
HYPRE_Int *num_ghost;
|
|
|
|
HYPRE_Int part;
|
|
HYPRE_Int nvars, var;
|
|
|
|
object_type= hypre_SStructVectorObjectType(vector);
|
|
|
|
HYPRE_SStructVectorCreate(hypre_SStructVectorComm(vector),
|
|
hypre_SStructVectorGrid(vector),
|
|
&new_vector);
|
|
HYPRE_SStructVectorSetObjectType(new_vector, object_type);
|
|
|
|
if (object_type == HYPRE_SSTRUCT || object_type == HYPRE_STRUCT)
|
|
{
|
|
for (part= 0; part< nparts; part++)
|
|
{
|
|
pvector = hypre_SStructVectorPVector(vector, part);
|
|
new_pvector= hypre_SStructVectorPVector(new_vector, part);
|
|
nvars = hypre_SStructPVectorNVars(pvector);
|
|
|
|
for (var= 0; var< nvars; var++)
|
|
{
|
|
svector= hypre_SStructPVectorSVector(pvector, var);
|
|
num_ghost= hypre_StructVectorNumGhost(svector);
|
|
|
|
new_svector= hypre_SStructPVectorSVector(new_pvector, var);
|
|
hypre_StructVectorSetNumGhost(new_svector, num_ghost);
|
|
}
|
|
}
|
|
}
|
|
|
|
HYPRE_SStructVectorInitialize(new_vector);
|
|
HYPRE_SStructVectorAssemble(new_vector);
|
|
|
|
return ( (void *) new_vector );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
void *
|
|
hypre_SStructKrylovCreateVectorArray(HYPRE_Int n, void *vvector )
|
|
{
|
|
hypre_SStructVector *vector = vvector;
|
|
hypre_SStructVector **new_vector;
|
|
HYPRE_Int object_type;
|
|
|
|
HYPRE_Int nparts= hypre_SStructVectorNParts(vector);
|
|
hypre_SStructPVector *pvector;
|
|
hypre_StructVector *svector;
|
|
hypre_SStructPVector *new_pvector;
|
|
hypre_StructVector *new_svector;
|
|
HYPRE_Int *num_ghost;
|
|
|
|
HYPRE_Int part;
|
|
HYPRE_Int nvars, var;
|
|
|
|
HYPRE_Int i;
|
|
|
|
object_type= hypre_SStructVectorObjectType(vector);
|
|
|
|
new_vector = hypre_CTAlloc(hypre_SStructVector*,n);
|
|
for (i=0; i < n; i++)
|
|
{
|
|
HYPRE_SStructVectorCreate(hypre_SStructVectorComm(vector),
|
|
hypre_SStructVectorGrid(vector),
|
|
&new_vector[i]);
|
|
HYPRE_SStructVectorSetObjectType(new_vector[i], object_type);
|
|
|
|
if (object_type == HYPRE_SSTRUCT || object_type == HYPRE_STRUCT)
|
|
{
|
|
for (part= 0; part< nparts; part++)
|
|
{
|
|
pvector = hypre_SStructVectorPVector(vector, part);
|
|
new_pvector= hypre_SStructVectorPVector(new_vector[i], part);
|
|
nvars = hypre_SStructPVectorNVars(pvector);
|
|
|
|
for (var= 0; var< nvars; var++)
|
|
{
|
|
svector= hypre_SStructPVectorSVector(pvector, var);
|
|
num_ghost= hypre_StructVectorNumGhost(svector);
|
|
|
|
new_svector= hypre_SStructPVectorSVector(new_pvector, var);
|
|
hypre_StructVectorSetNumGhost(new_svector, num_ghost);
|
|
}
|
|
}
|
|
}
|
|
|
|
HYPRE_SStructVectorInitialize(new_vector[i]);
|
|
HYPRE_SStructVectorAssemble(new_vector[i]);
|
|
}
|
|
|
|
return ( (void *) new_vector );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovDestroyVector( void *vvector )
|
|
{
|
|
hypre_SStructVector *vector = vvector;
|
|
|
|
return( HYPRE_SStructVectorDestroy( vector ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
void *
|
|
hypre_SStructKrylovMatvecCreate( void *A,
|
|
void *x )
|
|
{
|
|
void *matvec_data;
|
|
|
|
hypre_SStructMatvecCreate( &matvec_data );
|
|
hypre_SStructMatvecSetup( matvec_data,
|
|
(hypre_SStructMatrix *) A,
|
|
(hypre_SStructVector *) x );
|
|
|
|
return ( matvec_data );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovMatvec( void *matvec_data,
|
|
HYPRE_Complex alpha,
|
|
void *A,
|
|
void *x,
|
|
HYPRE_Complex beta,
|
|
void *y )
|
|
{
|
|
return ( hypre_SStructMatvec( alpha,
|
|
(hypre_SStructMatrix *) A,
|
|
(hypre_SStructVector *) x,
|
|
beta,
|
|
(hypre_SStructVector *) y ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovMatvecDestroy( void *matvec_data )
|
|
{
|
|
return ( hypre_SStructMatvecDestroy( matvec_data ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Real
|
|
hypre_SStructKrylovInnerProd( void *x,
|
|
void *y )
|
|
{
|
|
HYPRE_Real result;
|
|
|
|
hypre_SStructInnerProd( (hypre_SStructVector *) x,
|
|
(hypre_SStructVector *) y, &result );
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovCopyVector( void *x,
|
|
void *y )
|
|
{
|
|
return ( hypre_SStructCopy( (hypre_SStructVector *) x,
|
|
(hypre_SStructVector *) y ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovClearVector( void *x )
|
|
{
|
|
return ( hypre_SStructVectorSetConstantValues( (hypre_SStructVector *) x,
|
|
0.0 ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovScaleVector( HYPRE_Complex alpha,
|
|
void *x )
|
|
{
|
|
return ( hypre_SStructScale( alpha, (hypre_SStructVector *) x ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovAxpy( HYPRE_Complex alpha,
|
|
void *x,
|
|
void *y )
|
|
{
|
|
return ( hypre_SStructAxpy( alpha, (hypre_SStructVector *) x,
|
|
(hypre_SStructVector *) y ) );
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructKrylovCommInfo( void *A,
|
|
HYPRE_Int *my_id,
|
|
HYPRE_Int *num_procs )
|
|
{
|
|
MPI_Comm comm = hypre_SStructMatrixComm((hypre_SStructMatrix *) A);
|
|
hypre_MPI_Comm_size(comm,num_procs);
|
|
hypre_MPI_Comm_rank(comm,my_id);
|
|
return hypre_error_flag;
|
|
}
|
|
|