hypre/sstruct_mv/sstruct_axpy.c
falgout e3181f26b1 Added 64 bit feature using HYPRE_Int (see tracker [issue489] for details).
Changed MPI routines to hypre_MPI routines.
Added hypre_printf, etc. routines.
Added AUTOTEST tests to look for 'int' and 'MPI_' calls.
Added a new approach for the Fortran interface (not implemented everywhere yet).
2010-12-20 19:27:44 +00:00

91 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 "headers.h"
/*--------------------------------------------------------------------------
* hypre_SStructPAxpy
*--------------------------------------------------------------------------*/
HYPRE_Int
hypre_SStructPAxpy( double 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( double 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;
}