cannam@95
|
1 /*
|
cannam@95
|
2 * Copyright (c) 2003, 2007-11 Matteo Frigo
|
cannam@95
|
3 * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology
|
cannam@95
|
4 *
|
cannam@95
|
5 * This program is free software; you can redistribute it and/or modify
|
cannam@95
|
6 * it under the terms of the GNU General Public License as published by
|
cannam@95
|
7 * the Free Software Foundation; either version 2 of the License, or
|
cannam@95
|
8 * (at your option) any later version.
|
cannam@95
|
9 *
|
cannam@95
|
10 * This program is distributed in the hope that it will be useful,
|
cannam@95
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
cannam@95
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
cannam@95
|
13 * GNU General Public License for more details.
|
cannam@95
|
14 *
|
cannam@95
|
15 * You should have received a copy of the GNU General Public License
|
cannam@95
|
16 * along with this program; if not, write to the Free Software
|
cannam@95
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
cannam@95
|
18 *
|
cannam@95
|
19 */
|
cannam@95
|
20
|
cannam@95
|
21 #include "ifftw-mpi.h"
|
cannam@95
|
22
|
cannam@95
|
23 /* problem.c: */
|
cannam@95
|
24 typedef struct {
|
cannam@95
|
25 problem super;
|
cannam@95
|
26 dtensor *sz;
|
cannam@95
|
27 INT vn; /* vector length (vector stride 1) */
|
cannam@95
|
28 R *I, *O; /* contiguous interleaved arrays */
|
cannam@95
|
29
|
cannam@95
|
30 int sign; /* FFTW_FORWARD / FFTW_BACKWARD */
|
cannam@95
|
31 unsigned flags; /* TRANSPOSED_IN/OUT meaningful for rnk>1 only
|
cannam@95
|
32 SCRAMBLED_IN/OUT meaningful for 1d transforms only */
|
cannam@95
|
33
|
cannam@95
|
34 MPI_Comm comm;
|
cannam@95
|
35 } problem_mpi_dft;
|
cannam@95
|
36
|
cannam@95
|
37 problem *XM(mkproblem_dft)(const dtensor *sz, INT vn,
|
cannam@95
|
38 R *I, R *O, MPI_Comm comm,
|
cannam@95
|
39 int sign, unsigned flags);
|
cannam@95
|
40 problem *XM(mkproblem_dft_d)(dtensor *sz, INT vn,
|
cannam@95
|
41 R *I, R *O, MPI_Comm comm,
|
cannam@95
|
42 int sign, unsigned flags);
|
cannam@95
|
43
|
cannam@95
|
44 /* solve.c: */
|
cannam@95
|
45 void XM(dft_solve)(const plan *ego_, const problem *p_);
|
cannam@95
|
46
|
cannam@95
|
47 /* plans have same operands as rdft plans, so just re-use */
|
cannam@95
|
48 typedef plan_rdft plan_mpi_dft;
|
cannam@95
|
49 #define MKPLAN_MPI_DFT(type, adt, apply) \
|
cannam@95
|
50 (type *)X(mkplan_rdft)(sizeof(type), adt, apply)
|
cannam@95
|
51
|
cannam@95
|
52 int XM(dft_serial_applicable)(const problem_mpi_dft *p);
|
cannam@95
|
53
|
cannam@95
|
54 /* various solvers */
|
cannam@95
|
55 void XM(dft_rank_geq2_register)(planner *p);
|
cannam@95
|
56 void XM(dft_rank_geq2_transposed_register)(planner *p);
|
cannam@95
|
57 void XM(dft_serial_register)(planner *p);
|
cannam@95
|
58 void XM(dft_rank1_bigvec_register)(planner *p);
|
cannam@95
|
59 void XM(dft_rank1_register)(planner *p);
|