Chris@42: /* Chris@42: * Copyright (c) 2003, 2007-14 Matteo Frigo Chris@42: * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology Chris@42: * Chris@42: * This program is free software; you can redistribute it and/or modify Chris@42: * it under the terms of the GNU General Public License as published by Chris@42: * the Free Software Foundation; either version 2 of the License, or Chris@42: * (at your option) any later version. Chris@42: * Chris@42: * This program is distributed in the hope that it will be useful, Chris@42: * but WITHOUT ANY WARRANTY; without even the implied warranty of Chris@42: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Chris@42: * GNU General Public License for more details. Chris@42: * Chris@42: * You should have received a copy of the GNU General Public License Chris@42: * along with this program; if not, write to the Free Software Chris@42: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA Chris@42: * Chris@42: */ Chris@42: Chris@42: #include "ifftw-mpi.h" Chris@42: Chris@42: /* problem.c: */ Chris@42: typedef struct { Chris@42: problem super; Chris@42: dtensor *sz; Chris@42: INT vn; /* vector length (vector stride 1) */ Chris@42: R *I, *O; /* contiguous interleaved arrays */ Chris@42: Chris@42: int sign; /* FFTW_FORWARD / FFTW_BACKWARD */ Chris@42: unsigned flags; /* TRANSPOSED_IN/OUT meaningful for rnk>1 only Chris@42: SCRAMBLED_IN/OUT meaningful for 1d transforms only */ Chris@42: Chris@42: MPI_Comm comm; Chris@42: } problem_mpi_dft; Chris@42: Chris@42: problem *XM(mkproblem_dft)(const dtensor *sz, INT vn, Chris@42: R *I, R *O, MPI_Comm comm, Chris@42: int sign, unsigned flags); Chris@42: problem *XM(mkproblem_dft_d)(dtensor *sz, INT vn, Chris@42: R *I, R *O, MPI_Comm comm, Chris@42: int sign, unsigned flags); Chris@42: Chris@42: /* solve.c: */ Chris@42: void XM(dft_solve)(const plan *ego_, const problem *p_); Chris@42: Chris@42: /* plans have same operands as rdft plans, so just re-use */ Chris@42: typedef plan_rdft plan_mpi_dft; Chris@42: #define MKPLAN_MPI_DFT(type, adt, apply) \ Chris@42: (type *)X(mkplan_rdft)(sizeof(type), adt, apply) Chris@42: Chris@42: int XM(dft_serial_applicable)(const problem_mpi_dft *p); Chris@42: Chris@42: /* various solvers */ Chris@42: void XM(dft_rank_geq2_register)(planner *p); Chris@42: void XM(dft_rank_geq2_transposed_register)(planner *p); Chris@42: void XM(dft_serial_register)(planner *p); Chris@42: void XM(dft_rank1_bigvec_register)(planner *p); Chris@42: void XM(dft_rank1_register)(planner *p);