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 "api.h"
|
cannam@95
|
22
|
cannam@95
|
23 static plan *mkplan0(planner *plnr, unsigned flags,
|
cannam@95
|
24 const problem *prb, int hash_info,
|
cannam@95
|
25 wisdom_state_t wisdom_state)
|
cannam@95
|
26 {
|
cannam@95
|
27 /* map API flags into FFTW flags */
|
cannam@95
|
28 X(mapflags)(plnr, flags);
|
cannam@95
|
29
|
cannam@95
|
30 plnr->flags.hash_info = hash_info;
|
cannam@95
|
31 plnr->wisdom_state = wisdom_state;
|
cannam@95
|
32
|
cannam@95
|
33 /* create plan */
|
cannam@95
|
34 return plnr->adt->mkplan(plnr, prb);
|
cannam@95
|
35 }
|
cannam@95
|
36
|
cannam@95
|
37 static unsigned force_estimator(unsigned flags)
|
cannam@95
|
38 {
|
cannam@95
|
39 flags &= ~(FFTW_MEASURE | FFTW_PATIENT | FFTW_EXHAUSTIVE);
|
cannam@95
|
40 return (flags | FFTW_ESTIMATE);
|
cannam@95
|
41 }
|
cannam@95
|
42
|
cannam@95
|
43 static plan *mkplan(planner *plnr, unsigned flags,
|
cannam@95
|
44 const problem *prb, int hash_info)
|
cannam@95
|
45 {
|
cannam@95
|
46 plan *pln;
|
cannam@95
|
47
|
cannam@95
|
48 pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
|
cannam@95
|
49
|
cannam@95
|
50 if (plnr->wisdom_state == WISDOM_NORMAL && !pln) {
|
cannam@95
|
51 /* maybe the planner failed because of inconsistent wisdom;
|
cannam@95
|
52 plan again ignoring infeasible wisdom */
|
cannam@95
|
53 pln = mkplan0(plnr, force_estimator(flags), prb,
|
cannam@95
|
54 hash_info, WISDOM_IGNORE_INFEASIBLE);
|
cannam@95
|
55 }
|
cannam@95
|
56
|
cannam@95
|
57 if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
|
cannam@95
|
58 /* if the planner detected a wisdom inconsistency,
|
cannam@95
|
59 forget all wisdom and plan again */
|
cannam@95
|
60 plnr->adt->forget(plnr, FORGET_EVERYTHING);
|
cannam@95
|
61
|
cannam@95
|
62 A(!pln);
|
cannam@95
|
63 pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
|
cannam@95
|
64
|
cannam@95
|
65 if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
|
cannam@95
|
66 /* if it still fails, plan without wisdom */
|
cannam@95
|
67 plnr->adt->forget(plnr, FORGET_EVERYTHING);
|
cannam@95
|
68
|
cannam@95
|
69 A(!pln);
|
cannam@95
|
70 pln = mkplan0(plnr, force_estimator(flags),
|
cannam@95
|
71 prb, hash_info, WISDOM_IGNORE_ALL);
|
cannam@95
|
72 }
|
cannam@95
|
73 }
|
cannam@95
|
74
|
cannam@95
|
75 return pln;
|
cannam@95
|
76 }
|
cannam@95
|
77
|
cannam@95
|
78 apiplan *X(mkapiplan)(int sign, unsigned flags, problem *prb)
|
cannam@95
|
79 {
|
cannam@95
|
80 apiplan *p = 0;
|
cannam@95
|
81 plan *pln;
|
cannam@95
|
82 unsigned flags_used_for_planning;
|
cannam@95
|
83 planner *plnr = X(the_planner)();
|
cannam@95
|
84 unsigned int pats[] = {FFTW_ESTIMATE, FFTW_MEASURE,
|
cannam@95
|
85 FFTW_PATIENT, FFTW_EXHAUSTIVE};
|
cannam@95
|
86 int pat, pat_max;
|
cannam@95
|
87 double pcost = 0;
|
cannam@95
|
88
|
cannam@95
|
89 if (flags & FFTW_WISDOM_ONLY) {
|
cannam@95
|
90 /* Special mode that returns a plan only if wisdom is present,
|
cannam@95
|
91 and returns 0 otherwise. This is now documented in the manual,
|
cannam@95
|
92 as a way to detect whether wisdom is available for a problem. */
|
cannam@95
|
93 flags_used_for_planning = flags;
|
cannam@95
|
94 pln = mkplan0(plnr, flags, prb, 0, WISDOM_ONLY);
|
cannam@95
|
95 } else {
|
cannam@95
|
96 pat_max = flags & FFTW_ESTIMATE ? 0 :
|
cannam@95
|
97 (flags & FFTW_EXHAUSTIVE ? 3 :
|
cannam@95
|
98 (flags & FFTW_PATIENT ? 2 : 1));
|
cannam@95
|
99 pat = plnr->timelimit >= 0 ? 0 : pat_max;
|
cannam@95
|
100
|
cannam@95
|
101 flags &= ~(FFTW_ESTIMATE | FFTW_MEASURE |
|
cannam@95
|
102 FFTW_PATIENT | FFTW_EXHAUSTIVE);
|
cannam@95
|
103
|
cannam@95
|
104 plnr->start_time = X(get_crude_time)();
|
cannam@95
|
105
|
cannam@95
|
106 /* plan at incrementally increasing patience until we run
|
cannam@95
|
107 out of time */
|
cannam@95
|
108 for (pln = 0, flags_used_for_planning = 0; pat <= pat_max; ++pat) {
|
cannam@95
|
109 plan *pln1;
|
cannam@95
|
110 unsigned tmpflags = flags | pats[pat];
|
cannam@95
|
111 pln1 = mkplan(plnr, tmpflags, prb, 0);
|
cannam@95
|
112
|
cannam@95
|
113 if (!pln1) {
|
cannam@95
|
114 /* don't bother continuing if planner failed or timed out */
|
cannam@95
|
115 A(!pln || plnr->timed_out);
|
cannam@95
|
116 break;
|
cannam@95
|
117 }
|
cannam@95
|
118
|
cannam@95
|
119 X(plan_destroy_internal)(pln);
|
cannam@95
|
120 pln = pln1;
|
cannam@95
|
121 flags_used_for_planning = tmpflags;
|
cannam@95
|
122 pcost = pln->pcost;
|
cannam@95
|
123 }
|
cannam@95
|
124 }
|
cannam@95
|
125
|
cannam@95
|
126 if (pln) {
|
cannam@95
|
127 /* build apiplan */
|
cannam@95
|
128 p = (apiplan *) MALLOC(sizeof(apiplan), PLANS);
|
cannam@95
|
129 p->prb = prb;
|
cannam@95
|
130 p->sign = sign; /* cache for execute_dft */
|
cannam@95
|
131
|
cannam@95
|
132 /* re-create plan from wisdom, adding blessing */
|
cannam@95
|
133 p->pln = mkplan(plnr, flags_used_for_planning, prb, BLESSING);
|
cannam@95
|
134
|
cannam@95
|
135 /* record pcost from most recent measurement for use in X(cost) */
|
cannam@95
|
136 p->pln->pcost = pcost;
|
cannam@95
|
137
|
cannam@95
|
138 if (sizeof(trigreal) > sizeof(R)) {
|
cannam@95
|
139 /* this is probably faster, and we have enough trigreal
|
cannam@95
|
140 bits to maintain accuracy */
|
cannam@95
|
141 X(plan_awake)(p->pln, AWAKE_SQRTN_TABLE);
|
cannam@95
|
142 } else {
|
cannam@95
|
143 /* more accurate */
|
cannam@95
|
144 X(plan_awake)(p->pln, AWAKE_SINCOS);
|
cannam@95
|
145 }
|
cannam@95
|
146
|
cannam@95
|
147 /* we don't use pln for p->pln, above, since by re-creating the
|
cannam@95
|
148 plan we might use more patient wisdom from a timed-out mkplan */
|
cannam@95
|
149 X(plan_destroy_internal)(pln);
|
cannam@95
|
150 } else
|
cannam@95
|
151 X(problem_destroy)(prb);
|
cannam@95
|
152
|
cannam@95
|
153 /* discard all information not necessary to reconstruct the plan */
|
cannam@95
|
154 plnr->adt->forget(plnr, FORGET_ACCURSED);
|
cannam@95
|
155
|
cannam@95
|
156 #ifdef FFTW_RANDOM_ESTIMATOR
|
cannam@95
|
157 X(random_estimate_seed)++; /* subsequent "random" plans are distinct */
|
cannam@95
|
158 #endif
|
cannam@95
|
159
|
cannam@95
|
160 return p;
|
cannam@95
|
161 }
|
cannam@95
|
162
|
cannam@95
|
163 void X(destroy_plan)(X(plan) p)
|
cannam@95
|
164 {
|
cannam@95
|
165 if (p) {
|
cannam@95
|
166 X(plan_awake)(p->pln, SLEEPY);
|
cannam@95
|
167 X(plan_destroy_internal)(p->pln);
|
cannam@95
|
168 X(problem_destroy)(p->prb);
|
cannam@95
|
169 X(ifree)(p);
|
cannam@95
|
170 }
|
cannam@95
|
171 }
|