Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F81249060
test_sparse_solver_mumps.cc
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Wed, Sep 4, 23:26
Size
4 KB
Mime Type
text/x-c
Expires
Fri, Sep 6, 23:26 (2 d)
Engine
blob
Format
Raw Data
Handle
20458634
Attached To
rAKA akantu
test_sparse_solver_mumps.cc
View Options
/**
* @file test_sparse_solver_mumps.cc
*
* @author Nicolas Richart <nicolas.richart@epfl.ch>
*
* @date creation: Fri May 19 2017
* @date last modification: Wed Nov 08 2017
*
* @brief test the matrix vector product in parallel
*
*
* Copyright (©) 2016-2018 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* Akantu 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, either version 3 of the License, or (at your option) any
* later version.
*
* Akantu is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#include "aka_common.hh"
#include "dof_synchronizer.hh"
#include "element_synchronizer.hh"
#include "mesh.hh"
#include "mesh_accessor.hh"
#include "mesh_partition_scotch.hh"
#include "sparse_matrix_aij.hh"
#include "sparse_solver_mumps.hh"
#include "terms_to_assemble.hh"
/* -------------------------------------------------------------------------- */
#include <iostream>
/* -------------------------------------------------------------------------- */
using
namespace
akantu
;
/* -------------------------------------------------------------------------- */
void
genMesh
(
Mesh
&
mesh
,
UInt
nb_nodes
);
/* -------------------------------------------------------------------------- */
/* -------------------------------------------------------------------------- */
int
main
(
int
argc
,
char
*
argv
[])
{
initialize
(
argc
,
argv
);
const
UInt
spatial_dimension
=
1
;
const
UInt
nb_global_dof
=
11
;
const
auto
&
comm
=
Communicator
::
getStaticCommunicator
();
Int
psize
=
comm
.
getNbProc
();
Int
prank
=
comm
.
whoAmI
();
Mesh
mesh
(
spatial_dimension
);
if
(
prank
==
0
)
{
genMesh
(
mesh
,
nb_global_dof
);
RandomGenerator
<
UInt
>::
seed
(
1496137735
);
}
else
{
RandomGenerator
<
UInt
>::
seed
(
2992275470
);
}
mesh
.
distribute
();
UInt
node
=
0
;
for
(
auto
pos
:
mesh
.
getNodes
())
{
std
::
cout
<<
prank
<<
" "
<<
node
<<
" pos: "
<<
pos
<<
" ["
<<
mesh
.
getNodeGlobalId
(
node
)
<<
"] "
<<
mesh
.
getNodeFlag
(
node
)
<<
std
::
endl
;
++
node
;
}
UInt
nb_nodes
=
mesh
.
getNbNodes
();
DOFManagerDefault
dof_manager
(
mesh
,
"test_dof_manager"
);
Array
<
Real
>
x
(
nb_nodes
);
dof_manager
.
registerDOFs
(
"x"
,
x
,
_dst_nodal
);
const
auto
&
local_equation_number
=
dof_manager
.
getLocalEquationsNumbers
(
"x"
);
auto
&
A
=
dof_manager
.
getNewMatrix
(
"A"
,
_symmetric
);
Array
<
Real
>
b
(
nb_nodes
);
TermsToAssemble
terms
;
for
(
UInt
i
=
0
;
i
<
nb_nodes
;
++
i
)
{
if
(
dof_manager
.
isLocalOrMasterDOF
(
i
))
{
auto
li
=
local_equation_number
(
i
);
auto
gi
=
dof_manager
.
localToGlobalEquationNumber
(
li
);
terms
(
i
,
i
)
=
1.
/
(
1.
+
gi
);
}
}
dof_manager
.
assemblePreassembledMatrix
(
"x"
,
"x"
,
"A"
,
terms
);
std
::
stringstream
str
;
str
<<
"Matrix_"
<<
prank
<<
".mtx"
;
A
.
saveMatrix
(
str
.
str
());
for
(
UInt
n
=
0
;
n
<
nb_nodes
;
++
n
)
{
b
(
n
)
=
1.
;
}
SparseSolverMumps
solver
(
dof_manager
,
"A"
);
solver
.
solve
(
x
,
b
);
auto
&&
check
=
[
&
](
auto
&&
xs
)
{
debug
::
setDebugLevel
(
dblTest
);
std
::
cout
<<
xs
<<
std
::
endl
;
debug
::
setDebugLevel
(
dblWarning
);
UInt
d
=
1.
;
for
(
auto
x
:
xs
)
{
if
(
std
::
abs
(
x
-
d
)
/
d
>
1e-15
)
AKANTU_EXCEPTION
(
"Error in the solution: "
<<
x
<<
" != "
<<
d
<<
" ["
<<
(
std
::
abs
(
x
-
d
)
/
d
)
<<
"]."
);
++
d
;
}
};
if
(
psize
>
1
)
{
auto
&
sync
=
dynamic_cast
<
DOFManagerDefault
&>
(
dof_manager
).
getSynchronizer
();
if
(
prank
==
0
)
{
Array
<
Real
>
x_gathered
(
dof_manager
.
getSystemSize
());
sync
.
gather
(
x
,
x_gathered
);
check
(
x_gathered
);
}
else
{
sync
.
gather
(
x
);
}
}
else
{
check
(
x
);
}
finalize
();
return
0
;
}
/* -------------------------------------------------------------------------- */
void
genMesh
(
Mesh
&
mesh
,
UInt
nb_nodes
)
{
MeshAccessor
mesh_accessor
(
mesh
);
Array
<
Real
>
&
nodes
=
mesh_accessor
.
getNodes
();
Array
<
UInt
>
&
conn
=
mesh_accessor
.
getConnectivity
(
_segment_2
);
nodes
.
resize
(
nb_nodes
);
for
(
UInt
n
=
0
;
n
<
nb_nodes
;
++
n
)
{
nodes
(
n
,
_x
)
=
n
*
(
1.
/
(
nb_nodes
-
1
));
}
conn
.
resize
(
nb_nodes
-
1
);
for
(
UInt
n
=
0
;
n
<
nb_nodes
-
1
;
++
n
)
{
conn
(
n
,
0
)
=
n
;
conn
(
n
,
1
)
=
n
+
1
;
}
mesh_accessor
.
makeReady
();
}
Event Timeline
Log In to Comment