cannam@127
|
1 /*
|
cannam@127
|
2 * Copyright (c) 2003, 2007-14 Matteo Frigo
|
cannam@127
|
3 * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology
|
cannam@127
|
4 *
|
cannam@127
|
5 * This program is free software; you can redistribute it and/or modify
|
cannam@127
|
6 * it under the terms of the GNU General Public License as published by
|
cannam@127
|
7 * the Free Software Foundation; either version 2 of the License, or
|
cannam@127
|
8 * (at your option) any later version.
|
cannam@127
|
9 *
|
cannam@127
|
10 * This program is distributed in the hope that it will be useful,
|
cannam@127
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
cannam@127
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
cannam@127
|
13 * GNU General Public License for more details.
|
cannam@127
|
14 *
|
cannam@127
|
15 * You should have received a copy of the GNU General Public License
|
cannam@127
|
16 * along with this program; if not, write to the Free Software
|
cannam@127
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
cannam@127
|
18 *
|
cannam@127
|
19 */
|
cannam@127
|
20
|
cannam@127
|
21 #include "api.h"
|
cannam@127
|
22
|
cannam@127
|
23 static planner_hook_t before_planner_hook = 0, after_planner_hook = 0;
|
cannam@127
|
24
|
cannam@127
|
25 void X(set_planner_hooks)(planner_hook_t before, planner_hook_t after)
|
cannam@127
|
26 {
|
cannam@127
|
27 before_planner_hook = before;
|
cannam@127
|
28 after_planner_hook = after;
|
cannam@127
|
29 }
|
cannam@127
|
30
|
cannam@127
|
31 static plan *mkplan0(planner *plnr, unsigned flags,
|
cannam@127
|
32 const problem *prb, unsigned hash_info,
|
cannam@127
|
33 wisdom_state_t wisdom_state)
|
cannam@127
|
34 {
|
cannam@127
|
35 /* map API flags into FFTW flags */
|
cannam@127
|
36 X(mapflags)(plnr, flags);
|
cannam@127
|
37
|
cannam@127
|
38 plnr->flags.hash_info = hash_info;
|
cannam@127
|
39 plnr->wisdom_state = wisdom_state;
|
cannam@127
|
40
|
cannam@127
|
41 /* create plan */
|
cannam@127
|
42 return plnr->adt->mkplan(plnr, prb);
|
cannam@127
|
43 }
|
cannam@127
|
44
|
cannam@127
|
45 static unsigned force_estimator(unsigned flags)
|
cannam@127
|
46 {
|
cannam@127
|
47 flags &= ~(FFTW_MEASURE | FFTW_PATIENT | FFTW_EXHAUSTIVE);
|
cannam@127
|
48 return (flags | FFTW_ESTIMATE);
|
cannam@127
|
49 }
|
cannam@127
|
50
|
cannam@127
|
51 static plan *mkplan(planner *plnr, unsigned flags,
|
cannam@127
|
52 const problem *prb, unsigned hash_info)
|
cannam@127
|
53 {
|
cannam@127
|
54 plan *pln;
|
cannam@127
|
55
|
cannam@127
|
56 pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
|
cannam@127
|
57
|
cannam@127
|
58 if (plnr->wisdom_state == WISDOM_NORMAL && !pln) {
|
cannam@127
|
59 /* maybe the planner failed because of inconsistent wisdom;
|
cannam@127
|
60 plan again ignoring infeasible wisdom */
|
cannam@127
|
61 pln = mkplan0(plnr, force_estimator(flags), prb,
|
cannam@127
|
62 hash_info, WISDOM_IGNORE_INFEASIBLE);
|
cannam@127
|
63 }
|
cannam@127
|
64
|
cannam@127
|
65 if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
|
cannam@127
|
66 /* if the planner detected a wisdom inconsistency,
|
cannam@127
|
67 forget all wisdom and plan again */
|
cannam@127
|
68 plnr->adt->forget(plnr, FORGET_EVERYTHING);
|
cannam@127
|
69
|
cannam@127
|
70 A(!pln);
|
cannam@127
|
71 pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
|
cannam@127
|
72
|
cannam@127
|
73 if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
|
cannam@127
|
74 /* if it still fails, plan without wisdom */
|
cannam@127
|
75 plnr->adt->forget(plnr, FORGET_EVERYTHING);
|
cannam@127
|
76
|
cannam@127
|
77 A(!pln);
|
cannam@127
|
78 pln = mkplan0(plnr, force_estimator(flags),
|
cannam@127
|
79 prb, hash_info, WISDOM_IGNORE_ALL);
|
cannam@127
|
80 }
|
cannam@127
|
81 }
|
cannam@127
|
82
|
cannam@127
|
83 return pln;
|
cannam@127
|
84 }
|
cannam@127
|
85
|
cannam@127
|
86 apiplan *X(mkapiplan)(int sign, unsigned flags, problem *prb)
|
cannam@127
|
87 {
|
cannam@127
|
88 apiplan *p = 0;
|
cannam@127
|
89 plan *pln;
|
cannam@127
|
90 unsigned flags_used_for_planning;
|
cannam@127
|
91 planner *plnr;
|
cannam@127
|
92 static const unsigned int pats[] = {FFTW_ESTIMATE, FFTW_MEASURE,
|
cannam@127
|
93 FFTW_PATIENT, FFTW_EXHAUSTIVE};
|
cannam@127
|
94 int pat, pat_max;
|
cannam@127
|
95 double pcost = 0;
|
cannam@127
|
96
|
cannam@127
|
97 if (before_planner_hook)
|
cannam@127
|
98 before_planner_hook();
|
cannam@127
|
99
|
cannam@127
|
100 plnr = X(the_planner)();
|
cannam@127
|
101
|
cannam@127
|
102 if (flags & FFTW_WISDOM_ONLY) {
|
cannam@127
|
103 /* Special mode that returns a plan only if wisdom is present,
|
cannam@127
|
104 and returns 0 otherwise. This is now documented in the manual,
|
cannam@127
|
105 as a way to detect whether wisdom is available for a problem. */
|
cannam@127
|
106 flags_used_for_planning = flags;
|
cannam@127
|
107 pln = mkplan0(plnr, flags, prb, 0, WISDOM_ONLY);
|
cannam@127
|
108 } else {
|
cannam@127
|
109 pat_max = flags & FFTW_ESTIMATE ? 0 :
|
cannam@127
|
110 (flags & FFTW_EXHAUSTIVE ? 3 :
|
cannam@127
|
111 (flags & FFTW_PATIENT ? 2 : 1));
|
cannam@127
|
112 pat = plnr->timelimit >= 0 ? 0 : pat_max;
|
cannam@127
|
113
|
cannam@127
|
114 flags &= ~(FFTW_ESTIMATE | FFTW_MEASURE |
|
cannam@127
|
115 FFTW_PATIENT | FFTW_EXHAUSTIVE);
|
cannam@127
|
116
|
cannam@127
|
117 plnr->start_time = X(get_crude_time)();
|
cannam@127
|
118
|
cannam@127
|
119 /* plan at incrementally increasing patience until we run
|
cannam@127
|
120 out of time */
|
cannam@127
|
121 for (pln = 0, flags_used_for_planning = 0; pat <= pat_max; ++pat) {
|
cannam@127
|
122 plan *pln1;
|
cannam@127
|
123 unsigned tmpflags = flags | pats[pat];
|
cannam@127
|
124 pln1 = mkplan(plnr, tmpflags, prb, 0u);
|
cannam@127
|
125
|
cannam@127
|
126 if (!pln1) {
|
cannam@127
|
127 /* don't bother continuing if planner failed or timed out */
|
cannam@127
|
128 A(!pln || plnr->timed_out);
|
cannam@127
|
129 break;
|
cannam@127
|
130 }
|
cannam@127
|
131
|
cannam@127
|
132 X(plan_destroy_internal)(pln);
|
cannam@127
|
133 pln = pln1;
|
cannam@127
|
134 flags_used_for_planning = tmpflags;
|
cannam@127
|
135 pcost = pln->pcost;
|
cannam@127
|
136 }
|
cannam@127
|
137 }
|
cannam@127
|
138
|
cannam@127
|
139 if (pln) {
|
cannam@127
|
140 /* build apiplan */
|
cannam@127
|
141 p = (apiplan *) MALLOC(sizeof(apiplan), PLANS);
|
cannam@127
|
142 p->prb = prb;
|
cannam@127
|
143 p->sign = sign; /* cache for execute_dft */
|
cannam@127
|
144
|
cannam@127
|
145 /* re-create plan from wisdom, adding blessing */
|
cannam@127
|
146 p->pln = mkplan(plnr, flags_used_for_planning, prb, BLESSING);
|
cannam@127
|
147
|
cannam@127
|
148 /* record pcost from most recent measurement for use in X(cost) */
|
cannam@127
|
149 p->pln->pcost = pcost;
|
cannam@127
|
150
|
cannam@127
|
151 if (sizeof(trigreal) > sizeof(R)) {
|
cannam@127
|
152 /* this is probably faster, and we have enough trigreal
|
cannam@127
|
153 bits to maintain accuracy */
|
cannam@127
|
154 X(plan_awake)(p->pln, AWAKE_SQRTN_TABLE);
|
cannam@127
|
155 } else {
|
cannam@127
|
156 /* more accurate */
|
cannam@127
|
157 X(plan_awake)(p->pln, AWAKE_SINCOS);
|
cannam@127
|
158 }
|
cannam@127
|
159
|
cannam@127
|
160 /* we don't use pln for p->pln, above, since by re-creating the
|
cannam@127
|
161 plan we might use more patient wisdom from a timed-out mkplan */
|
cannam@127
|
162 X(plan_destroy_internal)(pln);
|
cannam@127
|
163 } else
|
cannam@127
|
164 X(problem_destroy)(prb);
|
cannam@127
|
165
|
cannam@127
|
166 /* discard all information not necessary to reconstruct the plan */
|
cannam@127
|
167 plnr->adt->forget(plnr, FORGET_ACCURSED);
|
cannam@127
|
168
|
cannam@127
|
169 #ifdef FFTW_RANDOM_ESTIMATOR
|
cannam@127
|
170 X(random_estimate_seed)++; /* subsequent "random" plans are distinct */
|
cannam@127
|
171 #endif
|
cannam@127
|
172
|
cannam@127
|
173 if (after_planner_hook)
|
cannam@127
|
174 after_planner_hook();
|
cannam@127
|
175
|
cannam@127
|
176 return p;
|
cannam@127
|
177 }
|
cannam@127
|
178
|
cannam@127
|
179 void X(destroy_plan)(X(plan) p)
|
cannam@127
|
180 {
|
cannam@127
|
181 if (p) {
|
cannam@127
|
182 X(plan_awake)(p->pln, SLEEPY);
|
cannam@127
|
183 X(plan_destroy_internal)(p->pln);
|
cannam@127
|
184 X(problem_destroy)(p->prb);
|
cannam@127
|
185 X(ifree)(p);
|
cannam@127
|
186 }
|
cannam@127
|
187 }
|
cannam@127
|
188
|
cannam@127
|
189 int X(alignment_of)(R *p)
|
cannam@127
|
190 {
|
cannam@127
|
191 return X(ialignment_of(p));
|
cannam@127
|
192 }
|