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.
120 lines
3.8 KiB
C
120 lines
3.8 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_struct_ls.h"
|
|
#include "temp_multivector.h"
|
|
|
|
HYPRE_Int
|
|
hypre_StructVectorSetRandomValues( hypre_StructVector *vector,
|
|
HYPRE_Int seed )
|
|
{
|
|
hypre_Box *v_data_box;
|
|
|
|
HYPRE_Int vi;
|
|
HYPRE_Real *vp;
|
|
|
|
hypre_BoxArray *boxes;
|
|
hypre_Box *box;
|
|
hypre_Index loop_size;
|
|
hypre_IndexRef start;
|
|
hypre_Index unit_stride;
|
|
|
|
HYPRE_Int i;
|
|
|
|
/*-----------------------------------------------------------------------
|
|
* Set the vector coefficients
|
|
*-----------------------------------------------------------------------*/
|
|
|
|
srand( seed );
|
|
|
|
hypre_SetIndex3(unit_stride, 1, 1, 1);
|
|
|
|
boxes = hypre_StructGridBoxes(hypre_StructVectorGrid(vector));
|
|
hypre_ForBoxI(i, boxes)
|
|
{
|
|
box = hypre_BoxArrayBox(boxes, i);
|
|
start = hypre_BoxIMin(box);
|
|
|
|
v_data_box =
|
|
hypre_BoxArrayBox(hypre_StructVectorDataSpace(vector), i);
|
|
vp = hypre_StructVectorBoxData(vector, i);
|
|
|
|
hypre_BoxGetSize(box, loop_size);
|
|
|
|
hypre_BoxLoop1Begin(hypre_StructVectorNDim(vector), loop_size,
|
|
v_data_box, start, unit_stride, vi);
|
|
#ifdef HYPRE_USING_OPENMP
|
|
#pragma omp parallel for private(HYPRE_BOX_PRIVATE,vi ) HYPRE_SMP_SCHEDULE
|
|
#endif
|
|
hypre_BoxLoop1For(vi)
|
|
{
|
|
vp[vi] = 2.0*rand()/RAND_MAX - 1.0;
|
|
}
|
|
hypre_BoxLoop1End(vi);
|
|
}
|
|
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
HYPRE_Int
|
|
hypre_StructSetRandomValues( void* v, HYPRE_Int seed ) {
|
|
|
|
return hypre_StructVectorSetRandomValues( (hypre_StructVector*)v, seed );
|
|
}
|
|
|
|
HYPRE_Int
|
|
HYPRE_StructSetupInterpreter( mv_InterfaceInterpreter *i )
|
|
{
|
|
i->CreateVector = hypre_StructKrylovCreateVector;
|
|
i->DestroyVector = hypre_StructKrylovDestroyVector;
|
|
i->InnerProd = hypre_StructKrylovInnerProd;
|
|
i->CopyVector = hypre_StructKrylovCopyVector;
|
|
i->ClearVector = hypre_StructKrylovClearVector;
|
|
i->SetRandomValues = hypre_StructSetRandomValues;
|
|
i->ScaleVector = hypre_StructKrylovScaleVector;
|
|
i->Axpy = hypre_StructKrylovAxpy;
|
|
|
|
i->CreateMultiVector = mv_TempMultiVectorCreateFromSampleVector;
|
|
i->CopyCreateMultiVector = mv_TempMultiVectorCreateCopy;
|
|
i->DestroyMultiVector = mv_TempMultiVectorDestroy;
|
|
|
|
i->Width = mv_TempMultiVectorWidth;
|
|
i->Height = mv_TempMultiVectorHeight;
|
|
i->SetMask = mv_TempMultiVectorSetMask;
|
|
i->CopyMultiVector = mv_TempMultiVectorCopy;
|
|
i->ClearMultiVector = mv_TempMultiVectorClear;
|
|
i->SetRandomVectors = mv_TempMultiVectorSetRandom;
|
|
i->MultiInnerProd = mv_TempMultiVectorByMultiVector;
|
|
i->MultiInnerProdDiag = mv_TempMultiVectorByMultiVectorDiag;
|
|
i->MultiVecMat = mv_TempMultiVectorByMatrix;
|
|
i->MultiVecMatDiag = mv_TempMultiVectorByDiagonal;
|
|
i->MultiAxpy = mv_TempMultiVectorAxpy;
|
|
i->MultiXapy = mv_TempMultiVectorXapy;
|
|
i->Eval = mv_TempMultiVectorEval;
|
|
|
|
return hypre_error_flag;
|
|
}
|
|
|
|
HYPRE_Int
|
|
HYPRE_StructSetupMatvec(HYPRE_MatvecFunctions * mv)
|
|
{
|
|
mv->MatvecCreate = hypre_StructKrylovMatvecCreate;
|
|
mv->Matvec = hypre_StructKrylovMatvec;
|
|
mv->MatvecDestroy = hypre_StructKrylovMatvecDestroy;
|
|
|
|
mv->MatMultiVecCreate = NULL;
|
|
mv->MatMultiVec = NULL;
|
|
mv->MatMultiVecDestroy = NULL;
|
|
|
|
return hypre_error_flag;
|
|
}
|