mirror of
https://github.com/cosmo-sims/monofonIC.git
synced 2024-09-19 17:03:45 +02:00
115 lines
2.1 KiB
C++
115 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include <logger.hh>
|
|
|
|
#include <complex>
|
|
|
|
#if defined(USE_MPI)
|
|
#include <mpi.h>
|
|
#include <fftw3-mpi.h>
|
|
#else
|
|
#include <fftw3.h>
|
|
#endif
|
|
|
|
#ifdef USE_SINGLEPRECISION
|
|
using real_t = float;
|
|
using complex_t = fftwf_complex;
|
|
#define FFTW_PREFIX fftwf
|
|
#else
|
|
using real_t = double;
|
|
using complex_t = fftw_complex;
|
|
#define FFTW_PREFIX fftw
|
|
#endif
|
|
|
|
using ccomplex_t = std::complex<real_t>;
|
|
|
|
#define FFTW_GEN_NAME_PRIM(a, b) a##_##b
|
|
#define FFTW_GEN_NAME(a, b) FFTW_GEN_NAME_PRIM(a, b)
|
|
#define FFTW_API(x) FFTW_GEN_NAME(FFTW_PREFIX, x)
|
|
|
|
using fftw_plan_t = FFTW_GEN_NAME(FFTW_PREFIX, plan);
|
|
|
|
#if defined(FFTW_MODE_PATIENT)
|
|
#define FFTW_RUNMODE FFTW_PATIENT
|
|
#elif defined(FFTW_MODE_MEASURE)
|
|
#define FFTW_RUNMODE FFTW_MEASURE
|
|
#else
|
|
#define FFTW_RUNMODE FFTW_ESTIMATE
|
|
#endif
|
|
|
|
#if defined(USE_MPI)
|
|
inline double get_wtime()
|
|
{
|
|
return MPI_Wtime();
|
|
}
|
|
|
|
inline int MPI_Get_rank( void ){
|
|
int rank, ret;
|
|
ret = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
assert( ret==MPI_SUCCESS );
|
|
return rank;
|
|
}
|
|
|
|
inline int MPI_Get_size( void ){
|
|
int size, ret;
|
|
ret = MPI_Comm_size(MPI_COMM_WORLD, &size);
|
|
assert( ret==MPI_SUCCESS );
|
|
return size;
|
|
}
|
|
|
|
template<typename T>
|
|
MPI_Datatype GetMPIDatatype( void )
|
|
{
|
|
if( typeid(T) == typeid(std::complex<float>) )
|
|
return MPI_COMPLEX;
|
|
|
|
if( typeid(T) == typeid(std::complex<double>) )
|
|
return MPI_DOUBLE_COMPLEX;
|
|
|
|
if( typeid(T) == typeid(int) )
|
|
return MPI_INT;
|
|
|
|
if( typeid(T) == typeid(unsigned) )
|
|
return MPI_UNSIGNED;
|
|
|
|
if( typeid(T) == typeid(float) )
|
|
return MPI_FLOAT;
|
|
|
|
if( typeid(T) == typeid(double) )
|
|
return MPI_DOUBLE;
|
|
|
|
if( typeid(T) == typeid(char) )
|
|
return MPI_CHAR;
|
|
|
|
abort();
|
|
|
|
}
|
|
|
|
|
|
#else
|
|
#if defined(_OPENMP)
|
|
#include <omp.h>
|
|
inline double get_wtime()
|
|
{
|
|
return omp_get_wtime();
|
|
}
|
|
#else
|
|
#include <ctime>
|
|
inline double get_wtime()
|
|
{
|
|
return std::clock() / double(CLOCKS_PER_SEC);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
|
|
|
|
namespace CONFIG
|
|
{
|
|
extern int MPI_thread_support;
|
|
extern int MPI_task_rank;
|
|
extern int MPI_task_size;
|
|
extern bool MPI_ok;
|
|
extern bool MPI_threads_ok;
|
|
extern bool FFTW_threads_ok;
|
|
} // namespace CONFIG
|