66 lines
1.7 KiB
C++
66 lines
1.7 KiB
C++
/* -*- c++ -*- ----------------------------------------------------------
|
|
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
|
|
https://www.lammps.org/, Sandia National Laboratories
|
|
LAMMPS development team: developers@lammps.org
|
|
|
|
Copyright (2003) Sandia Corporation. Under the terms of Contract
|
|
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
|
|
certain rights in this software. This software is distributed under
|
|
the GNU General Public License.
|
|
|
|
See the README file in the top-level LAMMPS directory.
|
|
------------------------------------------------------------------------- */
|
|
|
|
// common FFT library related defines and compilation settings
|
|
|
|
#ifndef LMP_FFT_SETTINGS_H
|
|
#define LMP_FFT_SETTINGS_H
|
|
|
|
// if a user sets FFTW, it means FFTW3
|
|
|
|
#ifdef FFT_FFTW
|
|
#ifndef FFT_FFTW3
|
|
#undef FFT_FFTW
|
|
#define FFT_FFTW3
|
|
#endif
|
|
#endif
|
|
|
|
// set strings for library info output
|
|
|
|
#if defined(FFT_HEFFTE)
|
|
#if defined(FFT_HEFFTE_FFTW)
|
|
#define LMP_FFT_LIB "HeFFTe(FFTW3)"
|
|
#elif defined(FFT_HEFFTE_MKL)
|
|
#define LMP_FFT_LIB "HeFFTe(MKL)"
|
|
#else
|
|
#define LMP_FFT_LIB "HeFFTe(builtin)"
|
|
#endif
|
|
#elif defined(FFT_FFTW3)
|
|
#define LMP_FFT_LIB "FFTW3"
|
|
#elif defined(FFT_MKL)
|
|
#define LMP_FFT_LIB "MKL FFT"
|
|
#elif defined(FFT_MKL_GPU)
|
|
#define LMP_FFT_LIB "MKL GPU FFT"
|
|
#elif defined(FFT_CUFFT)
|
|
#define LMP_FFT_LIB "cuFFT"
|
|
#elif defined(FFT_HIPFFT)
|
|
#define LMP_FFT_LIB "hipFFT"
|
|
#else
|
|
#define LMP_FFT_LIB "KISS FFT"
|
|
#endif
|
|
|
|
#ifdef FFT_SINGLE
|
|
typedef float FFT_SCALAR;
|
|
#define FFT_PRECISION 1
|
|
#define LMP_FFT_PREC "single"
|
|
#define MPI_FFT_SCALAR MPI_FLOAT
|
|
#else
|
|
|
|
typedef double FFT_SCALAR;
|
|
#define FFT_PRECISION 2
|
|
#define LMP_FFT_PREC "double"
|
|
#define MPI_FFT_SCALAR MPI_DOUBLE
|
|
#endif
|
|
|
|
#endif
|