Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F78379551
mpi_test_fft_engine.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
Tue, Aug 20, 11:22
Size
4 KB
Mime Type
text/x-c
Expires
Thu, Aug 22, 11:22 (2 d)
Engine
blob
Format
Raw Data
Handle
20036126
Attached To
rMUSPECTRE µSpectre
mpi_test_fft_engine.cc
View Options
/**
* @file mpi_test_fft_engine.cc
*
* @author Lars Pastewka <lars.pastewka@imtek.uni-freiburg.de>
*
* @date 06 Mar 2017
*
* @brief tests for MPI-parallel fft engine implementations
*
* Copyright © 2017 Till Junge
*
* µSpectre is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation, either version 3, or (at
* your option) any later version.
*
* µSpectre 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
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Emacs; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#include <boost/mpl/list.hpp>
#include "tests.hh"
#include "mpi_context.hh"
#include "fft/fftwmpi_engine.hh"
#include "common/ccoord_operations.hh"
#include "common/field_collection.hh"
#include "common/field_map.hh"
#include "common/iterators.hh"
namespace
muSpectre
{
BOOST_AUTO_TEST_SUITE
(
fftwmpi_engine
);
/* ---------------------------------------------------------------------- */
template
<
Dim_t
DimS
,
Dim_t
DimM
,
Dim_t
resolution
>
struct
FFTW_fixture
{
constexpr
static
Dim_t
box_resolution
{
resolution
};
constexpr
static
Real
box_length
{
4.5
};
constexpr
static
Dim_t
sdim
{
DimS
};
constexpr
static
Dim_t
mdim
{
DimM
};
constexpr
static
Ccoord_t
<
sdim
>
res
()
{
return
CcoordOps
::
get_cube
<
DimS
>
(
box_resolution
);
}
FFTW_fixture
()
:
engine
(
res
(),
CcoordOps
::
get_cube
<
DimS
>
(
box_length
),
µ
Spectre
::
MPIContext
::
get_context
().
comm
)
{}
FFTWMPIEngine
<
DimS
,
DimM
>
engine
;
};
struct
FFTW_fixture_python_segfault
{
constexpr
static
Dim_t
dim
{
twoD
};
constexpr
static
Dim_t
sdim
{
twoD
};
constexpr
static
Dim_t
mdim
{
twoD
};
constexpr
static
Ccoord_t
<
sdim
>
res
()
{
return
{
6
,
4
};}
FFTW_fixture_python_segfault
()
:
engine
{
res
(),
{
3.
,
3
},
µ
Spectre
::
MPIContext
::
get_context
().
comm
}
{}
FFTWMPIEngine
<
sdim
,
mdim
>
engine
;
};
using
fixlist
=
boost
::
mpl
::
list
<
FFTW_fixture
<
twoD
,
twoD
,
3
>
,
FFTW_fixture
<
twoD
,
threeD
,
3
>
,
FFTW_fixture
<
threeD
,
threeD
,
3
>
,
FFTW_fixture
<
twoD
,
twoD
,
4
>
,
FFTW_fixture
<
twoD
,
threeD
,
4
>
,
FFTW_fixture
<
threeD
,
threeD
,
4
>
,
FFTW_fixture_python_segfault
>
;
/* ---------------------------------------------------------------------- */
BOOST_FIXTURE_TEST_CASE_TEMPLATE
(
Constructor_test
,
Fix
,
fixlist
,
Fix
)
{
BOOST_CHECK_NO_THROW
(
Fix
::
engine
.
initialise
(
FFT_PlanFlags
::
estimate
));
BOOST_CHECK_EQUAL
(
Fix
::
engine
.
size
(),
CcoordOps
::
get_size
(
Fix
::
res
()));
}
#if 0
/* ---------------------------------------------------------------------- */
BOOST_FIXTURE_TEST_CASE_TEMPLATE(fft_test, Fix, fixlist, Fix) {
Fix::engine.initialise(FFT_PlanFlags::estimate);
constexpr Dim_t order{2};
using FC_t = GlobalFieldCollection<Fix::sdim, Fix::mdim>;
FC_t fc;
auto & input{make_field<TensorField<FC_t, Real, order, Fix::mdim>>("input", fc)};
auto & ref {make_field<TensorField<FC_t, Real, order, Fix::mdim>>("reference", fc)};
auto & result{make_field<TensorField<FC_t, Real, order, Fix::mdim>>("result", fc)};
fc.initialise(Fix::res());
using map_t = MatrixFieldMap<FC_t, Real, Fix::mdim, Fix::mdim>;
map_t inmap{input};
auto refmap{map_t{ref}};
auto resultmap{map_t{result}};
size_t cntr{0};
for (auto tup: akantu::zip(inmap, refmap)) {
cntr++;
auto & in_{std::get<0>(tup)};
auto & ref_{std::get<1>(tup)};
in_.setRandom();
ref_ = in_;
}
auto & complex_field = Fix::engine.fft(input);
using cmap_t = MatrixFieldMap<LocalFieldCollection<Fix::sdim, Fix::mdim>, Complex, Fix::mdim, Fix::mdim>;
cmap_t complex_map(complex_field);
Real error = complex_map[0].imag().norm();
BOOST_CHECK_LT(error, tol);
/* make sure, the engine has not modified input (which is
unfortunately const-casted internally, hence this test) */
for (auto && tup: akantu::zip(inmap, refmap)) {
Real error{(std::get<0>(tup) - std::get<1>(tup)).norm()};
BOOST_CHECK_LT(error, tol);
}
/* make sure that the ifft of fft returns the original*/
Fix::engine.ifft(result);
for (auto && tup: akantu::zip(resultmap, refmap)) {
Real error{(std::get<0>(tup)*Fix::engine.normalisation() - std::get<1>(tup)).norm()};
BOOST_CHECK_LT(error, tol);
if (error > tol) {
std::cout << std::get<0>(tup).array()/std::get<1>(tup).array() << std::endl << std::endl;
}
}
}
#endif
BOOST_AUTO_TEST_SUITE_END
();
}
// muSpectre
Event Timeline
Log In to Comment