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*/