Actual source code: chwirut2.c
1: /*
2: Include "petsctao.h" so that we can use TAO solvers. Note that this
3: file automatically includes libraries such as:
4: petsc.h - base PETSc routines petscvec.h - vectors
5: petscsys.h - system routines petscmat.h - matrices
6: petscis.h - index sets petscksp.h - Krylov subspace methods
7: petscviewer.h - viewers petscpc.h - preconditioners
9: */
11: #include <petsctao.h>
13: /*
14: Description: These data are the result of a NIST study involving
15: ultrasonic calibration. The response variable is
16: ultrasonic response, and the predictor variable is
17: metal distance.
19: Reference: Chwirut, D., NIST (197?).
20: Ultrasonic Reference Block Study.
21: */
23: static char help[]="Finds the nonlinear least-squares solution to the model \n\
24: y = exp[-b1*x]/(b2+b3*x) + e \n";
26: #define NOBSERVATIONS 214
27: #define NPARAMETERS 3
29: #define DIE_TAG 2000
30: #define IDLE_TAG 1000
32: /* User-defined application context */
33: typedef struct {
34: /* Working space */
35: PetscReal t[NOBSERVATIONS]; /* array of independent variables of observation */
36: PetscReal y[NOBSERVATIONS]; /* array of dependent variables */
37: PetscMPIInt size,rank;
38: } AppCtx;
40: /* User provided Routines */
41: PetscErrorCode InitializeData(AppCtx *user);
42: PetscErrorCode FormStartingPoint(Vec);
43: PetscErrorCode EvaluateFunction(Tao, Vec, Vec, void *);
44: PetscErrorCode TaskWorker(AppCtx *user);
45: PetscErrorCode StopWorkers(AppCtx *user);
46: PetscErrorCode RunSimulation(PetscReal *x, PetscInt i, PetscReal*f, AppCtx *user);
48: /*--------------------------------------------------------------------*/
49: int main(int argc,char **argv)
50: {
51: Vec x, f; /* solution, function */
52: Tao tao; /* Tao solver context */
53: AppCtx user; /* user-defined work context */
55: /* Initialize TAO and PETSc */
56: PetscInitialize(&argc,&argv,(char *)0,help);
57: MPI_Comm_size(MPI_COMM_WORLD,&user.size);
58: MPI_Comm_rank(MPI_COMM_WORLD,&user.rank);
59: InitializeData(&user);
61: /* Run optimization on rank 0 */
62: if (user.rank == 0) {
63: /* Allocate vectors */
64: VecCreateSeq(PETSC_COMM_SELF,NPARAMETERS,&x);
65: VecCreateSeq(PETSC_COMM_SELF,NOBSERVATIONS,&f);
67: /* TAO code begins here */
69: /* Create TAO solver and set desired solution method */
70: TaoCreate(PETSC_COMM_SELF,&tao);
71: TaoSetType(tao,TAOPOUNDERS);
73: /* Set the function and Jacobian routines. */
74: FormStartingPoint(x);
75: TaoSetSolution(tao,x);
76: TaoSetResidualRoutine(tao,f,EvaluateFunction,(void*)&user);
78: /* Check for any TAO command line arguments */
79: TaoSetFromOptions(tao);
81: /* Perform the Solve */
82: TaoSolve(tao);
84: /* Free TAO data structures */
85: TaoDestroy(&tao);
87: /* Free PETSc data structures */
88: VecDestroy(&x);
89: VecDestroy(&f);
90: StopWorkers(&user);
91: } else {
92: TaskWorker(&user);
93: }
94: PetscFinalize();
95: return 0;
96: }
98: /*--------------------------------------------------------------------*/
99: PetscErrorCode EvaluateFunction(Tao tao, Vec X, Vec F, void *ptr)
100: {
101: AppCtx *user = (AppCtx *)ptr;
102: PetscInt i;
103: PetscReal *x,*f;
105: VecGetArray(X,&x);
106: VecGetArray(F,&f);
107: if (user->size == 1) {
108: /* Single processor */
109: for (i=0;i<NOBSERVATIONS;i++) {
110: RunSimulation(x,i,&f[i],user);
111: }
112: } else {
113: /* Multiprocessor main */
114: PetscMPIInt tag;
115: PetscInt finishedtasks,next_task,checkedin;
116: PetscReal f_i=0.0;
117: MPI_Status status;
119: next_task=0;
120: finishedtasks=0;
121: checkedin=0;
123: while (finishedtasks < NOBSERVATIONS || checkedin < user->size-1) {
124: MPI_Recv(&f_i,1,MPIU_REAL,MPI_ANY_SOURCE,MPI_ANY_TAG,PETSC_COMM_WORLD,&status);
125: if (status.MPI_TAG == IDLE_TAG) {
126: checkedin++;
127: } else {
129: tag = status.MPI_TAG;
130: f[tag] = (PetscReal)f_i;
131: finishedtasks++;
132: }
134: if (next_task<NOBSERVATIONS) {
135: MPI_Send(x,NPARAMETERS,MPIU_REAL,status.MPI_SOURCE,next_task,PETSC_COMM_WORLD);
136: next_task++;
138: } else {
139: /* Send idle message */
140: MPI_Send(x,NPARAMETERS,MPIU_REAL,status.MPI_SOURCE,IDLE_TAG,PETSC_COMM_WORLD);
141: }
142: }
143: }
144: VecRestoreArray(X,&x);
145: VecRestoreArray(F,&f);
146: PetscLogFlops(6*NOBSERVATIONS);
147: return 0;
148: }
150: /* ------------------------------------------------------------ */
151: PetscErrorCode FormStartingPoint(Vec X)
152: {
153: PetscReal *x;
155: VecGetArray(X,&x);
156: x[0] = 0.15;
157: x[1] = 0.008;
158: x[2] = 0.010;
159: VecRestoreArray(X,&x);
160: return 0;
161: }
163: /* ---------------------------------------------------------------------- */
164: PetscErrorCode InitializeData(AppCtx *user)
165: {
166: PetscReal *t=user->t,*y=user->y;
167: PetscInt i=0;
169: y[i] = 92.9000; t[i++] = 0.5000;
170: y[i] = 78.7000; t[i++] = 0.6250;
171: y[i] = 64.2000; t[i++] = 0.7500;
172: y[i] = 64.9000; t[i++] = 0.8750;
173: y[i] = 57.1000; t[i++] = 1.0000;
174: y[i] = 43.3000; t[i++] = 1.2500;
175: y[i] = 31.1000; t[i++] = 1.7500;
176: y[i] = 23.6000; t[i++] = 2.2500;
177: y[i] = 31.0500; t[i++] = 1.7500;
178: y[i] = 23.7750; t[i++] = 2.2500;
179: y[i] = 17.7375; t[i++] = 2.7500;
180: y[i] = 13.8000; t[i++] = 3.2500;
181: y[i] = 11.5875; t[i++] = 3.7500;
182: y[i] = 9.4125; t[i++] = 4.2500;
183: y[i] = 7.7250; t[i++] = 4.7500;
184: y[i] = 7.3500; t[i++] = 5.2500;
185: y[i] = 8.0250; t[i++] = 5.7500;
186: y[i] = 90.6000; t[i++] = 0.5000;
187: y[i] = 76.9000; t[i++] = 0.6250;
188: y[i] = 71.6000; t[i++] = 0.7500;
189: y[i] = 63.6000; t[i++] = 0.8750;
190: y[i] = 54.0000; t[i++] = 1.0000;
191: y[i] = 39.2000; t[i++] = 1.2500;
192: y[i] = 29.3000; t[i++] = 1.7500;
193: y[i] = 21.4000; t[i++] = 2.2500;
194: y[i] = 29.1750; t[i++] = 1.7500;
195: y[i] = 22.1250; t[i++] = 2.2500;
196: y[i] = 17.5125; t[i++] = 2.7500;
197: y[i] = 14.2500; t[i++] = 3.2500;
198: y[i] = 9.4500; t[i++] = 3.7500;
199: y[i] = 9.1500; t[i++] = 4.2500;
200: y[i] = 7.9125; t[i++] = 4.7500;
201: y[i] = 8.4750; t[i++] = 5.2500;
202: y[i] = 6.1125; t[i++] = 5.7500;
203: y[i] = 80.0000; t[i++] = 0.5000;
204: y[i] = 79.0000; t[i++] = 0.6250;
205: y[i] = 63.8000; t[i++] = 0.7500;
206: y[i] = 57.2000; t[i++] = 0.8750;
207: y[i] = 53.2000; t[i++] = 1.0000;
208: y[i] = 42.5000; t[i++] = 1.2500;
209: y[i] = 26.8000; t[i++] = 1.7500;
210: y[i] = 20.4000; t[i++] = 2.2500;
211: y[i] = 26.8500; t[i++] = 1.7500;
212: y[i] = 21.0000; t[i++] = 2.2500;
213: y[i] = 16.4625; t[i++] = 2.7500;
214: y[i] = 12.5250; t[i++] = 3.2500;
215: y[i] = 10.5375; t[i++] = 3.7500;
216: y[i] = 8.5875; t[i++] = 4.2500;
217: y[i] = 7.1250; t[i++] = 4.7500;
218: y[i] = 6.1125; t[i++] = 5.2500;
219: y[i] = 5.9625; t[i++] = 5.7500;
220: y[i] = 74.1000; t[i++] = 0.5000;
221: y[i] = 67.3000; t[i++] = 0.6250;
222: y[i] = 60.8000; t[i++] = 0.7500;
223: y[i] = 55.5000; t[i++] = 0.8750;
224: y[i] = 50.3000; t[i++] = 1.0000;
225: y[i] = 41.0000; t[i++] = 1.2500;
226: y[i] = 29.4000; t[i++] = 1.7500;
227: y[i] = 20.4000; t[i++] = 2.2500;
228: y[i] = 29.3625; t[i++] = 1.7500;
229: y[i] = 21.1500; t[i++] = 2.2500;
230: y[i] = 16.7625; t[i++] = 2.7500;
231: y[i] = 13.2000; t[i++] = 3.2500;
232: y[i] = 10.8750; t[i++] = 3.7500;
233: y[i] = 8.1750; t[i++] = 4.2500;
234: y[i] = 7.3500; t[i++] = 4.7500;
235: y[i] = 5.9625; t[i++] = 5.2500;
236: y[i] = 5.6250; t[i++] = 5.7500;
237: y[i] = 81.5000; t[i++] = .5000;
238: y[i] = 62.4000; t[i++] = .7500;
239: y[i] = 32.5000; t[i++] = 1.5000;
240: y[i] = 12.4100; t[i++] = 3.0000;
241: y[i] = 13.1200; t[i++] = 3.0000;
242: y[i] = 15.5600; t[i++] = 3.0000;
243: y[i] = 5.6300; t[i++] = 6.0000;
244: y[i] = 78.0000; t[i++] = .5000;
245: y[i] = 59.9000; t[i++] = .7500;
246: y[i] = 33.2000; t[i++] = 1.5000;
247: y[i] = 13.8400; t[i++] = 3.0000;
248: y[i] = 12.7500; t[i++] = 3.0000;
249: y[i] = 14.6200; t[i++] = 3.0000;
250: y[i] = 3.9400; t[i++] = 6.0000;
251: y[i] = 76.8000; t[i++] = .5000;
252: y[i] = 61.0000; t[i++] = .7500;
253: y[i] = 32.9000; t[i++] = 1.5000;
254: y[i] = 13.8700; t[i++] = 3.0000;
255: y[i] = 11.8100; t[i++] = 3.0000;
256: y[i] = 13.3100; t[i++] = 3.0000;
257: y[i] = 5.4400; t[i++] = 6.0000;
258: y[i] = 78.0000; t[i++] = .5000;
259: y[i] = 63.5000; t[i++] = .7500;
260: y[i] = 33.8000; t[i++] = 1.5000;
261: y[i] = 12.5600; t[i++] = 3.0000;
262: y[i] = 5.6300; t[i++] = 6.0000;
263: y[i] = 12.7500; t[i++] = 3.0000;
264: y[i] = 13.1200; t[i++] = 3.0000;
265: y[i] = 5.4400; t[i++] = 6.0000;
266: y[i] = 76.8000; t[i++] = .5000;
267: y[i] = 60.0000; t[i++] = .7500;
268: y[i] = 47.8000; t[i++] = 1.0000;
269: y[i] = 32.0000; t[i++] = 1.5000;
270: y[i] = 22.2000; t[i++] = 2.0000;
271: y[i] = 22.5700; t[i++] = 2.0000;
272: y[i] = 18.8200; t[i++] = 2.5000;
273: y[i] = 13.9500; t[i++] = 3.0000;
274: y[i] = 11.2500; t[i++] = 4.0000;
275: y[i] = 9.0000; t[i++] = 5.0000;
276: y[i] = 6.6700; t[i++] = 6.0000;
277: y[i] = 75.8000; t[i++] = .5000;
278: y[i] = 62.0000; t[i++] = .7500;
279: y[i] = 48.8000; t[i++] = 1.0000;
280: y[i] = 35.2000; t[i++] = 1.5000;
281: y[i] = 20.0000; t[i++] = 2.0000;
282: y[i] = 20.3200; t[i++] = 2.0000;
283: y[i] = 19.3100; t[i++] = 2.5000;
284: y[i] = 12.7500; t[i++] = 3.0000;
285: y[i] = 10.4200; t[i++] = 4.0000;
286: y[i] = 7.3100; t[i++] = 5.0000;
287: y[i] = 7.4200; t[i++] = 6.0000;
288: y[i] = 70.5000; t[i++] = .5000;
289: y[i] = 59.5000; t[i++] = .7500;
290: y[i] = 48.5000; t[i++] = 1.0000;
291: y[i] = 35.8000; t[i++] = 1.5000;
292: y[i] = 21.0000; t[i++] = 2.0000;
293: y[i] = 21.6700; t[i++] = 2.0000;
294: y[i] = 21.0000; t[i++] = 2.5000;
295: y[i] = 15.6400; t[i++] = 3.0000;
296: y[i] = 8.1700; t[i++] = 4.0000;
297: y[i] = 8.5500; t[i++] = 5.0000;
298: y[i] = 10.1200; t[i++] = 6.0000;
299: y[i] = 78.0000; t[i++] = .5000;
300: y[i] = 66.0000; t[i++] = .6250;
301: y[i] = 62.0000; t[i++] = .7500;
302: y[i] = 58.0000; t[i++] = .8750;
303: y[i] = 47.7000; t[i++] = 1.0000;
304: y[i] = 37.8000; t[i++] = 1.2500;
305: y[i] = 20.2000; t[i++] = 2.2500;
306: y[i] = 21.0700; t[i++] = 2.2500;
307: y[i] = 13.8700; t[i++] = 2.7500;
308: y[i] = 9.6700; t[i++] = 3.2500;
309: y[i] = 7.7600; t[i++] = 3.7500;
310: y[i] = 5.4400; t[i++] = 4.2500;
311: y[i] = 4.8700; t[i++] = 4.7500;
312: y[i] = 4.0100; t[i++] = 5.2500;
313: y[i] = 3.7500; t[i++] = 5.7500;
314: y[i] = 24.1900; t[i++] = 3.0000;
315: y[i] = 25.7600; t[i++] = 3.0000;
316: y[i] = 18.0700; t[i++] = 3.0000;
317: y[i] = 11.8100; t[i++] = 3.0000;
318: y[i] = 12.0700; t[i++] = 3.0000;
319: y[i] = 16.1200; t[i++] = 3.0000;
320: y[i] = 70.8000; t[i++] = .5000;
321: y[i] = 54.7000; t[i++] = .7500;
322: y[i] = 48.0000; t[i++] = 1.0000;
323: y[i] = 39.8000; t[i++] = 1.5000;
324: y[i] = 29.8000; t[i++] = 2.0000;
325: y[i] = 23.7000; t[i++] = 2.5000;
326: y[i] = 29.6200; t[i++] = 2.0000;
327: y[i] = 23.8100; t[i++] = 2.5000;
328: y[i] = 17.7000; t[i++] = 3.0000;
329: y[i] = 11.5500; t[i++] = 4.0000;
330: y[i] = 12.0700; t[i++] = 5.0000;
331: y[i] = 8.7400; t[i++] = 6.0000;
332: y[i] = 80.7000; t[i++] = .5000;
333: y[i] = 61.3000; t[i++] = .7500;
334: y[i] = 47.5000; t[i++] = 1.0000;
335: y[i] = 29.0000; t[i++] = 1.5000;
336: y[i] = 24.0000; t[i++] = 2.0000;
337: y[i] = 17.7000; t[i++] = 2.5000;
338: y[i] = 24.5600; t[i++] = 2.0000;
339: y[i] = 18.6700; t[i++] = 2.5000;
340: y[i] = 16.2400; t[i++] = 3.0000;
341: y[i] = 8.7400; t[i++] = 4.0000;
342: y[i] = 7.8700; t[i++] = 5.0000;
343: y[i] = 8.5100; t[i++] = 6.0000;
344: y[i] = 66.7000; t[i++] = .5000;
345: y[i] = 59.2000; t[i++] = .7500;
346: y[i] = 40.8000; t[i++] = 1.0000;
347: y[i] = 30.7000; t[i++] = 1.5000;
348: y[i] = 25.7000; t[i++] = 2.0000;
349: y[i] = 16.3000; t[i++] = 2.5000;
350: y[i] = 25.9900; t[i++] = 2.0000;
351: y[i] = 16.9500; t[i++] = 2.5000;
352: y[i] = 13.3500; t[i++] = 3.0000;
353: y[i] = 8.6200; t[i++] = 4.0000;
354: y[i] = 7.2000; t[i++] = 5.0000;
355: y[i] = 6.6400; t[i++] = 6.0000;
356: y[i] = 13.6900; t[i++] = 3.0000;
357: y[i] = 81.0000; t[i++] = .5000;
358: y[i] = 64.5000; t[i++] = .7500;
359: y[i] = 35.5000; t[i++] = 1.5000;
360: y[i] = 13.3100; t[i++] = 3.0000;
361: y[i] = 4.8700; t[i++] = 6.0000;
362: y[i] = 12.9400; t[i++] = 3.0000;
363: y[i] = 5.0600; t[i++] = 6.0000;
364: y[i] = 15.1900; t[i++] = 3.0000;
365: y[i] = 14.6200; t[i++] = 3.0000;
366: y[i] = 15.6400; t[i++] = 3.0000;
367: y[i] = 25.5000; t[i++] = 1.7500;
368: y[i] = 25.9500; t[i++] = 1.7500;
369: y[i] = 81.7000; t[i++] = .5000;
370: y[i] = 61.6000; t[i++] = .7500;
371: y[i] = 29.8000; t[i++] = 1.7500;
372: y[i] = 29.8100; t[i++] = 1.7500;
373: y[i] = 17.1700; t[i++] = 2.7500;
374: y[i] = 10.3900; t[i++] = 3.7500;
375: y[i] = 28.4000; t[i++] = 1.7500;
376: y[i] = 28.6900; t[i++] = 1.7500;
377: y[i] = 81.3000; t[i++] = .5000;
378: y[i] = 60.9000; t[i++] = .7500;
379: y[i] = 16.6500; t[i++] = 2.7500;
380: y[i] = 10.0500; t[i++] = 3.7500;
381: y[i] = 28.9000; t[i++] = 1.7500;
382: y[i] = 28.9500; t[i++] = 1.7500;
383: return 0;
384: }
386: PetscErrorCode TaskWorker(AppCtx *user)
387: {
388: PetscReal x[NPARAMETERS],f = 0.0;
389: PetscMPIInt tag=IDLE_TAG;
390: PetscInt index;
391: MPI_Status status;
393: /* Send check-in message to rank-0 */
395: MPI_Send(&f,1,MPIU_REAL,0,IDLE_TAG,PETSC_COMM_WORLD);
396: while (tag != DIE_TAG) {
397: MPI_Recv(x,NPARAMETERS,MPIU_REAL,0,MPI_ANY_TAG,PETSC_COMM_WORLD,&status);
398: tag = status.MPI_TAG;
399: if (tag == IDLE_TAG) {
400: MPI_Send(&f,1,MPIU_REAL,0,IDLE_TAG,PETSC_COMM_WORLD);
401: } else if (tag != DIE_TAG) {
402: index = (PetscInt)tag;
403: RunSimulation(x,index,&f,user);
404: MPI_Send(&f,1,MPIU_REAL,0,tag,PETSC_COMM_WORLD);
405: }
406: }
407: return 0;
408: }
410: PetscErrorCode RunSimulation(PetscReal *x, PetscInt i, PetscReal*f, AppCtx *user)
411: {
412: PetscReal *t = user->t;
413: PetscReal *y = user->y;
414: #if defined(PETSC_USE_REAL_SINGLE)
415: *f = y[i] - exp(-x[0]*t[i])/(x[1] + x[2]*t[i]); /* expf() for single-precision breaks this example on Freebsd, Valgrind errors on Linux */
416: #else
417: *f = y[i] - PetscExpScalar(-x[0]*t[i])/(x[1] + x[2]*t[i]);
418: #endif
419: return(0);
420: }
422: PetscErrorCode StopWorkers(AppCtx *user)
423: {
424: PetscInt checkedin;
425: MPI_Status status;
426: PetscReal f,x[NPARAMETERS];
428: checkedin=0;
429: while (checkedin < user->size-1) {
430: MPI_Recv(&f,1,MPIU_REAL,MPI_ANY_SOURCE,MPI_ANY_TAG,PETSC_COMM_WORLD,&status);
431: checkedin++;
432: PetscArrayzero(x,NPARAMETERS);
433: MPI_Send(x,NPARAMETERS,MPIU_REAL,status.MPI_SOURCE,DIE_TAG,PETSC_COMM_WORLD);
434: }
435: return 0;
436: }
438: /*TEST
440: build:
441: requires: !complex
443: test:
444: nsize: 3
445: requires: !single
446: args: -tao_smonitor -tao_max_it 100 -tao_type pounders -tao_gatol 1.e-5
448: TEST*/