Chris@82: /* Chris@82: * Copyright (c) 2003, 2007-14 Matteo Frigo Chris@82: * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology Chris@82: * Chris@82: * This program is free software; you can redistribute it and/or modify Chris@82: * it under the terms of the GNU General Public License as published by Chris@82: * the Free Software Foundation; either version 2 of the License, or Chris@82: * (at your option) any later version. Chris@82: * Chris@82: * This program is distributed in the hope that it will be useful, Chris@82: * but WITHOUT ANY WARRANTY; without even the implied warranty of Chris@82: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the Chris@82: * GNU General Public License for more details. Chris@82: * Chris@82: * You should have received a copy of the GNU General Public License Chris@82: * along with this program; if not, write to the Free Software Chris@82: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA Chris@82: * Chris@82: */ Chris@82: Chris@82: #include "mpi-rdft.h" Chris@82: Chris@82: static void destroy(problem *ego_) Chris@82: { Chris@82: problem_mpi_rdft *ego = (problem_mpi_rdft *) ego_; Chris@82: XM(dtensor_destroy)(ego->sz); Chris@82: MPI_Comm_free(&ego->comm); Chris@82: #if !defined(STRUCT_HACK_C99) && !defined(STRUCT_HACK_KR) Chris@82: X(ifree0)(ego->kind); Chris@82: #endif Chris@82: X(ifree)(ego_); Chris@82: } Chris@82: Chris@82: static void hash(const problem *p_, md5 *m) Chris@82: { Chris@82: const problem_mpi_rdft *p = (const problem_mpi_rdft *) p_; Chris@82: int i; Chris@82: X(md5puts)(m, "mpi-dft"); Chris@82: X(md5int)(m, p->I == p->O); Chris@82: /* don't include alignment -- may differ between processes Chris@82: X(md5int)(m, X(ialignment_of)(p->I)); Chris@82: X(md5int)(m, X(ialignment_of)(p->O)); Chris@82: ... note that applicability of MPI plans does not depend Chris@82: on alignment (although optimality may, in principle). */ Chris@82: XM(dtensor_md5)(m, p->sz); Chris@82: X(md5INT)(m, p->vn); Chris@82: for (i = 0; i < p->sz->rnk; ++i) Chris@82: X(md5int)(m, p->kind[i]); Chris@82: X(md5int)(m, p->flags); Chris@82: MPI_Comm_size(p->comm, &i); X(md5int)(m, i); Chris@82: A(XM(md5_equal)(*m, p->comm)); Chris@82: } Chris@82: Chris@82: static void print(const problem *ego_, printer *p) Chris@82: { Chris@82: const problem_mpi_rdft *ego = (const problem_mpi_rdft *) ego_; Chris@82: int i; Chris@82: p->print(p, "(mpi-rdft %d %d %d ", Chris@82: ego->I == ego->O, Chris@82: X(ialignment_of)(ego->I), Chris@82: X(ialignment_of)(ego->O)); Chris@82: XM(dtensor_print)(ego->sz, p); Chris@82: for (i = 0; i < ego->sz->rnk; ++i) Chris@82: p->print(p, " %d", (int)ego->kind[i]); Chris@82: p->print(p, " %D %d", ego->vn, ego->flags); Chris@82: MPI_Comm_size(ego->comm, &i); p->print(p, " %d)", i); Chris@82: } Chris@82: Chris@82: static void zero(const problem *ego_) Chris@82: { Chris@82: const problem_mpi_rdft *ego = (const problem_mpi_rdft *) ego_; Chris@82: R *I = ego->I; Chris@82: INT i, N; Chris@82: int my_pe; Chris@82: Chris@82: MPI_Comm_rank(ego->comm, &my_pe); Chris@82: N = ego->vn * XM(total_block)(ego->sz, IB, my_pe); Chris@82: for (i = 0; i < N; ++i) I[i] = K(0.0); Chris@82: } Chris@82: Chris@82: static const problem_adt padt = Chris@82: { Chris@82: PROBLEM_MPI_RDFT, Chris@82: hash, Chris@82: zero, Chris@82: print, Chris@82: destroy Chris@82: }; Chris@82: Chris@82: problem *XM(mkproblem_rdft)(const dtensor *sz, INT vn, Chris@82: R *I, R *O, Chris@82: MPI_Comm comm, Chris@82: const rdft_kind *kind, unsigned flags) Chris@82: { Chris@82: problem_mpi_rdft *ego; Chris@82: int i, rnk = sz->rnk; Chris@82: int n_pes; Chris@82: Chris@82: A(XM(dtensor_validp)(sz) && FINITE_RNK(sz->rnk)); Chris@82: MPI_Comm_size(comm, &n_pes); Chris@82: A(n_pes >= XM(num_blocks_total)(sz, IB) Chris@82: && n_pes >= XM(num_blocks_total)(sz, OB)); Chris@82: A(vn >= 0); Chris@82: Chris@82: #if defined(STRUCT_HACK_KR) Chris@82: ego = (problem_mpi_rdft *) X(mkproblem)(sizeof(problem_mpi_rdft) Chris@82: + sizeof(rdft_kind) Chris@82: * (rnk > 0 ? rnk - 1 : 0), &padt); Chris@82: #elif defined(STRUCT_HACK_C99) Chris@82: ego = (problem_mpi_rdft *) X(mkproblem)(sizeof(problem_mpi_rdft) Chris@82: + sizeof(rdft_kind) * rnk, &padt); Chris@82: #else Chris@82: ego = (problem_mpi_rdft *) X(mkproblem)(sizeof(problem_mpi_rdft), &padt); Chris@82: ego->kind = (rdft_kind *) MALLOC(sizeof(rdft_kind) * rnk, PROBLEMS); Chris@82: #endif Chris@82: Chris@82: /* enforce pointer equality if untainted pointers are equal */ Chris@82: if (UNTAINT(I) == UNTAINT(O)) Chris@82: I = O = JOIN_TAINT(I, O); Chris@82: Chris@82: ego->sz = XM(dtensor_canonical)(sz, 0); Chris@82: ego->vn = vn; Chris@82: ego->I = I; Chris@82: ego->O = O; Chris@82: for (i = 0; i< ego->sz->rnk; ++i) Chris@82: ego->kind[i] = kind[i]; Chris@82: Chris@82: /* canonicalize: replace TRANSPOSED_IN with TRANSPOSED_OUT by Chris@82: swapping the first two dimensions (for rnk > 1) */ Chris@82: if ((flags & TRANSPOSED_IN) && ego->sz->rnk > 1) { Chris@82: rdft_kind k = ego->kind[0]; Chris@82: ddim dim0 = ego->sz->dims[0]; Chris@82: ego->sz->dims[0] = ego->sz->dims[1]; Chris@82: ego->sz->dims[1] = dim0; Chris@82: ego->kind[0] = ego->kind[1]; Chris@82: ego->kind[1] = k; Chris@82: flags &= ~TRANSPOSED_IN; Chris@82: flags ^= TRANSPOSED_OUT; Chris@82: } Chris@82: ego->flags = flags; Chris@82: Chris@82: MPI_Comm_dup(comm, &ego->comm); Chris@82: Chris@82: return &(ego->super); Chris@82: } Chris@82: Chris@82: problem *XM(mkproblem_rdft_d)(dtensor *sz, INT vn, Chris@82: R *I, R *O, Chris@82: MPI_Comm comm, Chris@82: const rdft_kind *kind, unsigned flags) Chris@82: { Chris@82: problem *p = XM(mkproblem_rdft)(sz, vn, I, O, comm, kind, flags); Chris@82: XM(dtensor_destroy)(sz); Chris@82: return p; Chris@82: }