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.
88 lines
2.7 KiB
C
88 lines
2.7 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 axpy routine
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#include "_hypre_sstruct_mv.h"
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_SStructPAxpy
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructPAxpy( HYPRE_Complex alpha,
|
|
hypre_SStructPVector *px,
|
|
hypre_SStructPVector *py )
|
|
{
|
|
HYPRE_Int nvars = hypre_SStructPVectorNVars(px);
|
|
HYPRE_Int var;
|
|
|
|
for (var = 0; var < nvars; var++)
|
|
{
|
|
hypre_StructAxpy(alpha,
|
|
hypre_SStructPVectorSVector(px, var),
|
|
hypre_SStructPVectorSVector(py, var));
|
|
}
|
|
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_SStructAxpy
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
HYPRE_Int
|
|
hypre_SStructAxpy( HYPRE_Complex alpha,
|
|
hypre_SStructVector *x,
|
|
hypre_SStructVector *y )
|
|
{
|
|
HYPRE_Int nparts = hypre_SStructVectorNParts(x);
|
|
HYPRE_Int part;
|
|
|
|
HYPRE_Int x_object_type= hypre_SStructVectorObjectType(x);
|
|
HYPRE_Int y_object_type= hypre_SStructVectorObjectType(y);
|
|
|
|
if (x_object_type != y_object_type)
|
|
{
|
|
hypre_error_in_arg(2);
|
|
hypre_error_in_arg(3);
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
if (x_object_type == HYPRE_SSTRUCT)
|
|
{
|
|
for (part = 0; part < nparts; part++)
|
|
{
|
|
hypre_SStructPAxpy(alpha,
|
|
hypre_SStructVectorPVector(x, part),
|
|
hypre_SStructVectorPVector(y, part));
|
|
}
|
|
}
|
|
|
|
else if (x_object_type == HYPRE_PARCSR)
|
|
{
|
|
hypre_ParVector *x_par;
|
|
hypre_ParVector *y_par;
|
|
|
|
hypre_SStructVectorConvert(x, &x_par);
|
|
hypre_SStructVectorConvert(y, &y_par);
|
|
|
|
hypre_ParVectorAxpy(alpha, x_par, y_par);
|
|
}
|
|
|
|
return hypre_error_flag;
|
|
}
|