src/gromacs/gmxlib/nrnb.c

   1 /*
   2  *
   3  *                This source code is part of
   4  *
   5  *                 G   R   O   M   A   C   S
   6  *
   7  *          GROningen MAchine for Chemical Simulations
   8  *
   9  *                        VERSION 3.2.0
  10  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
  11  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
  12  * Copyright (c) 2001-2004, The GROMACS development team,
  13  * check out http://www.gromacs.org for more information.
  14
  15  * This program is free software; you can redistribute it and/or
  16  * modify it under the terms of the GNU General Public License
  17  * as published by the Free Software Foundation; either version 2
  18  * of the License, or (at your option) any later version.
  19  *
  20  * If you want to redistribute modifications, please consider that
  21  * scientific software is very special. Version control is crucial -
  22  * bugs must be traceable. We will be happy to consider code for
  23  * inclusion in the official distribution, but derived work must not
  24  * be called official GROMACS. Details are found in the README & COPYING
  25  * files - if they are missing, get the official version at www.gromacs.org.
  26  *
  27  * To help us fund GROMACS development, we humbly ask that you cite
  28  * the papers on the package - you can find them in the top README file.
  29  *
  30  * For more info, check our website at http://www.gromacs.org
  31  *
  32  * And Hey:
  33  * GROningen Mixture of Alchemy and Childrens' Stories
  34  */
  35 #ifdef HAVE_CONFIG_H
  36 #include <config.h>
  37 #endif
  38
  39 #include <string.h>
  40 #include "sysstuff.h"
  41 #include "gmx_fatal.h"
  42 #include "names.h"
  43 #include "macros.h"
  44 #include "nrnb.h"
  45 #include "main.h"
  46 #include "smalloc.h"
  47 #include "copyrite.h"
  48
  49 typedef struct {
  50   const char *name;
  51   int  flop;
  52 } t_nrnb_data;
  53
  54
  55 static const t_nrnb_data nbdata[eNRNB] = {
  56     { "LJ",                             33 }, /* nb_kernel010 */
  57     { "Buckingham",                     61 }, /* nb_kernel020 */
  58     { "VdW(T)",                         54 }, /* nb_kernel030 */
  59     { "Coulomb",                        27 }, /* nb_kernel100 */
  60     { "Coulomb [W3]",                   80 }, /* nb_kernel101 */
  61     { "Coulomb [W3-W3]",               234 }, /* nb_kernel102 */
  62     { "Coulomb [W4]",                   80 }, /* nb_kernel103 */
  63     { "Coulomb [W4-W4]",               234 }, /* nb_kernel104 */
  64     { "Coulomb + LJ",                   38 }, /* nb_kernel110 */
  65     { "Coulomb + LJ [W3]",              91 }, /* nb_kernel111 */
  66     { "Coulomb + LJ [W3-W3]",          245 }, /* nb_kernel112 */
  67     { "Coulomb + LJ [W4]",             113 }, /* nb_kernel113 */
  68     { "Coulomb + LJ [W4-W4]",          267 }, /* nb_kernel114 */
  69     { "Coulomb + Bham ",                64 }, /* nb_kernel120 */
  70     { "Coulomb + Bham [W3]",           117 }, /* nb_kernel121 */
  71     { "Coulomb + Bham [W3-W3]",        271 }, /* nb_kernel122 */
  72     { "Coulomb + Bham [W4]",           141 }, /* nb_kernel123 */
  73     { "Coulomb + Bham [W4-W4]",        295 }, /* nb_kernel124 */
  74     { "Coulomb + VdW(T) ",              59 }, /* nb_kernel130 */
  75     { "Coulomb + VdW(T) [W3]",         112 }, /* nb_kernel131 */
  76     { "Coulomb + VdW(T) [W3-W3]",      266 }, /* nb_kernel132 */
  77     { "Coulomb + VdW(T) [W4]",         134 }, /* nb_kernel133 */
  78     { "Coulomb + VdW(T) [W4-W4]",      288 }, /* nb_kernel134 */
  79     { "RF Coul",                        33 }, /* nb_kernel200 */
  80     { "RF Coul [W3]",                   98 }, /* nb_kernel201 */
  81     { "RF Coul [W3-W3]",               288 }, /* nb_kernel202 */
  82     { "RF Coul [W4]",                   98 }, /* nb_kernel203 */
  83     { "RF Coul [W4-W4]",               288 }, /* nb_kernel204 */
  84     { "RF Coul + LJ",                   44 }, /* nb_kernel210 */
  85     { "RF Coul + LJ [W3]",             109 }, /* nb_kernel211 */
  86     { "RF Coul + LJ [W3-W3]",          299 }, /* nb_kernel212 */
  87     { "RF Coul + LJ [W4]",             131 }, /* nb_kernel213 */
  88     { "RF Coul + LJ [W4-W4]",          321 }, /* nb_kernel214 */
  89     { "RF Coul + Bham ",                70 }, /* nb_kernel220 */
  90     { "RF Coul + Bham [W3]",           135 }, /* nb_kernel221 */
  91     { "RF Coul + Bham [W3-W3]",        325 }, /* nb_kernel222 */
  92     { "RF Coul + Bham [W4]",           159 }, /* nb_kernel223 */
  93     { "RF Coul + Bham [W4-W4]",        349 }, /* nb_kernel224 */
  94     { "RF Coul + VdW(T) ",              65 }, /* nb_kernel230 */
  95     { "RF Coul + VdW(T) [W3]",         130 }, /* nb_kernel231 */
  96     { "RF Coul + VdW(T) [W3-W3]",      320 }, /* nb_kernel232 */
  97     { "RF Coul + VdW(T) [W4]",         152 }, /* nb_kernel233 */
  98     { "RF Coul + VdW(T) [W4-W4]",      342 }, /* nb_kernel234 */
  99     { "Coul(T)",                        42 }, /* nb_kernel300 */
 100     { "Coul(T) [W3]",                  125 }, /* nb_kernel301 */
 101     { "Coul(T) [W3-W3]",               369 }, /* nb_kernel302 */
 102     { "Coul(T) [W4]",                  125 }, /* nb_kernel303 */
 103     { "Coul(T) [W4-W4]",               369 }, /* nb_kernel304 */
 104     { "Coul(T) + LJ",                   55 }, /* nb_kernel310 */
 105     { "Coul(T) + LJ [W3]",             138 }, /* nb_kernel311 */
 106     { "Coul(T) + LJ [W3-W3]",          382 }, /* nb_kernel312 */
 107     { "Coul(T) + LJ [W4]",             158 }, /* nb_kernel313 */
 108     { "Coul(T) + LJ [W4-W4]",          402 }, /* nb_kernel314 */
 109     { "Coul(T) + Bham",                 81 }, /* nb_kernel320 */
 110     { "Coul(T) + Bham [W3]",           164 }, /* nb_kernel321 */
 111     { "Coul(T) + Bham [W3-W3]",        408 }, /* nb_kernel322 */
 112     { "Coul(T) + Bham [W4]",           186 }, /* nb_kernel323 */
 113     { "Coul(T) + Bham [W4-W4]",        430 }, /* nb_kernel324 */
 114     { "Coul(T) + VdW(T)",               68 }, /* nb_kernel330 */
 115     { "Coul(T) + VdW(T) [W3]",         151 }, /* nb_kernel331 */
 116     { "Coul(T) + VdW(T) [W3-W3]",      395 }, /* nb_kernel332 */
 117     { "Coul(T) + VdW(T) [W4]",         179 }, /* nb_kernel333 */
 118     { "Coul(T) + VdW(T) [W4-W4]",      423 }, /* nb_kernel334 */
 119     { "Generalized Born Coulomb",       48 }, /* nb_kernel400 */
 120     { "GB Coulomb + LJ",                61 }, /* nb_kernel410 */
 121     { "GB Coulomb + VdW(T)",            79 }, /* nb_kernel430 */
 122     { "LJ NF",                          19 }, /* nb_kernel010nf */
 123     { "Buckingham NF",                  48 }, /* nb_kernel020nf */
 124     { "VdW(T) NF",                      33 }, /* nb_kernel030nf */
 125     { "Coulomb NF",                     16 }, /* nb_kernel100nf */
 126     { "Coulomb [W3] NF",                47 }, /* nb_kernel101nf */
 127     { "Coulomb [W3-W3] NF",            135 }, /* nb_kernel102nf */
 128     { "Coulomb [W4] NF",                47 }, /* nb_kernel103nf */
 129     { "Coulomb [W4-W4] NF",            135 }, /* nb_kernel104nf */
 130     { "Coulomb + LJ NF",                24 }, /* nb_kernel110nf */
 131     { "Coulomb + LJ [W3] NF",           55 }, /* nb_kernel111nf */
 132     { "Coulomb + LJ [W3-W3] NF",       143 }, /* nb_kernel112nf */
 133     { "Coulomb + LJ [W4] NF",           66 }, /* nb_kernel113nf */
 134     { "Coulomb + LJ [W4-W4] NF",       154 }, /* nb_kernel114nf */
 135     { "Coulomb + Bham  NF",             51 }, /* nb_kernel120nf */
 136     { "Coulomb + Bham [W3] NF",         82 }, /* nb_kernel121nf */
 137     { "Coulomb + Bham [W3-W3] NF",     170 }, /* nb_kernel122nf */
 138     { "Coulomb + Bham [W4] NF",         95 }, /* nb_kernel123nf */
 139     { "Coulomb + Bham [W4-W4] NF",     183 }, /* nb_kernel124nf */
 140     { "Coulomb + VdW(T)  NF",           36 }, /* nb_kernel130nf */
 141     { "Coulomb + VdW(T) [W3] NF",       67 }, /* nb_kernel131nf */
 142     { "Coulomb + VdW(T) [W3-W3] NF",   155 }, /* nb_kernel132nf */
 143     { "Coulomb + VdW(T) [W4] NF",       80 }, /* nb_kernel133nf */
 144     { "Coulomb + VdW(T) [W4-W4] NF",   168 }, /* nb_kernel134nf */
 145     { "RF Coul NF",                     19 }, /* nb_kernel200nf */
 146     { "RF Coul [W3] NF",                56 }, /* nb_kernel201nf */
 147     { "RF Coul [W3-W3] NF",            162 }, /* nb_kernel202nf */
 148     { "RF Coul [W4] NF",                56 }, /* nb_kernel203nf */
 149     { "RF Coul [W4-W4] NF",            162 }, /* nb_kernel204nf */
 150     { "RF Coul + LJ NF",                27 }, /* nb_kernel210nf */
 151     { "RF Coul + LJ [W3] NF",           64 }, /* nb_kernel211nf */
 152     { "RF Coul + LJ [W3-W3] NF",       170 }, /* nb_kernel212nf */
 153     { "RF Coul + LJ [W4] NF",           75 }, /* nb_kernel213nf */
 154     { "RF Coul + LJ [W4-W4] NF",       181 }, /* nb_kernel214nf */
 155     { "RF Coul + Bham  NF",             54 }, /* nb_kernel220nf */
 156     { "RF Coul + Bham [W3] NF",         91 }, /* nb_kernel221nf */
 157     { "RF Coul + Bham [W3-W3] NF",     197 }, /* nb_kernel222nf */
 158     { "RF Coul + Bham [W4] NF",        104 }, /* nb_kernel223nf */
 159     { "RF Coul + Bham [W4-W4] NF",     210 }, /* nb_kernel224nf */
 160     { "RF Coul + VdW(T)  NF",           39 }, /* nb_kernel230nf */
 161     { "RF Coul + VdW(T) [W3] NF",       76 }, /* nb_kernel231nf */
 162     { "RF Coul + VdW(T) [W3-W3] NF",   182 }, /* nb_kernel232nf */
 163     { "RF Coul + VdW(T) [W4] NF",       89 }, /* nb_kernel233nf */
 164     { "RF Coul + VdW(T) [W4-W4] NF",   195 }, /* nb_kernel234nf */
 165     { "Coul(T) NF",                     26 }, /* nb_kernel300nf */
 166     { "Coul(T) [W3] NF",                77 }, /* nb_kernel301nf */
 167     { "Coul(T) [W3-W3] NF",            225 }, /* nb_kernel302nf */
 168     { "Coul(T) [W4] NF",                77 }, /* nb_kernel303nf */
 169     { "Coul(T) [W4-W4] NF",            225 }, /* nb_kernel304nf */
 170     { "Coul(T) + LJ NF",                34 }, /* nb_kernel310nf */
 171     { "Coul(T) + LJ [W3] NF",           85 }, /* nb_kernel311nf */
 172     { "Coul(T) + LJ [W3-W3] NF",       233 }, /* nb_kernel312nf */
 173     { "Coul(T) + LJ [W4] NF",           96 }, /* nb_kernel313nf */
 174     { "Coul(T) + LJ [W4-W4] NF",       244 }, /* nb_kernel314nf */
 175     { "Coul(T) + Bham NF",              61 }, /* nb_kernel320nf */
 176     { "Coul(T) + Bham [W3] NF",        112 }, /* nb_kernel321nf */
 177     { "Coul(T) + Bham [W3-W3] NF",     260 }, /* nb_kernel322nf */
 178     { "Coul(T) + Bham [W4] NF",        125 }, /* nb_kernel323nf */
 179     { "Coul(T) + Bham [W4-W4] NF",     273 }, /* nb_kernel324nf */
 180     { "Coul(T) + VdW(T) NF",            42 }, /* nb_kernel330nf */
 181     { "Coul(T) + VdW(T) [W3] NF",       93 }, /* nb_kernel331nf */
 182     { "Coul(T) + VdW(T) [W3-W3] NF",   241 }, /* nb_kernel332nf */
 183     { "Coul(T) + VdW(T) [W4] NF",      110 }, /* nb_kernel333nf */
 184     { "Coul(T) + VdW(T) [W4-W4] NF",   258 }, /* nb_kernel334nf */
 185     { "Generalized Born Coulomb NF",    29 }, /* nb_kernel400nf */
 186     { "GB Coulomb + LJ NF",             37 }, /* nb_kernel410nf */
 187     { "GB Coulomb + VdW(T) NF",         49 }, /* nb_kernel430nf */
 188     { "Free energy innerloop",         150 }, /* free energy, estimate */
 189     { "All-vs-All, Coul + LJ",          38 },
 190     { "All-vs-All, GB + LJ",            61 },
 191     { "Outer nonbonded loop",           10 },
 192     { "1,4 nonbonded interactions",     90 },
 193     { "Born radii (Still)",             47 },
 194     { "Born radii (HCT/OBC)",          183 },
 195     { "Born force chain rule",          15 },
 196     { "All-vs-All Still radii",         47 },
 197     { "All-vs-All HCT/OBC radii",      183 },
 198     { "All-vs-All Born chain rule",     15 },
 199     { "Calc Weights",                   36 },
 200     { "Spread Q",                        6 },
 201     { "Spread Q Bspline",                2 },
 202     { "Gather F",                      23  },
 203     { "Gather F Bspline",              6   },
 204     { "3D-FFT",                        8   },
 205     { "Convolution",                   4   },
 206     { "Solve PME",                     64  },
 207     { "NS-Pairs",                      21  },
 208     { "Reset In Box",                  3   },
 209     { "Shift-X",                       6   },
 210     { "CG-CoM",                        3   },
 211     { "Sum Forces",                    1   },
 212     { "Bonds",                         59  },
 213     { "G96Bonds",                      44  },
 214     { "FENE Bonds",                    58  },
 215     { "Tab. Bonds",                    62  },
 216     { "Restraint Potential",           86  },
 217     { "Linear Angles",                 57  },
 218     { "Angles",                        168 },
 219     { "G96Angles",                     150 },
 220     { "Quartic Angles",                160 },
 221     { "Tab. Angles",                   169 },
 222     { "Propers",                       229 },
 223     { "Impropers",                     208 },
 224     { "RB-Dihedrals",                  247 },
 225     { "Four. Dihedrals",               247 },
 226     { "Tab. Dihedrals",                227 },
 227     { "Dist. Restr.",                  200 },
 228     { "Orient. Restr.",                200 },
 229     { "Dihedral Restr.",               200 },
 230     { "Pos. Restr.",                   50  },
 231     { "Flat-bottom posres",            50  },
 232     { "Angle Restr.",                  191 },
 233     { "Angle Restr. Z",                164 },
 234     { "Morse Potent.",                 83  },
 235     { "Cubic Bonds",                   54  },
 236     { "Walls",                         31  },
 237     { "Polarization",                  59  },
 238     { "Anharmonic Polarization",       72  },
 239     { "Water Pol.",                    62  },
 240     { "Thole Pol.",                    296 },
 241     { "Virial",                        18  },
 242     { "Update",                        31  },
 243     { "Ext.ens. Update",               54  },
 244     { "Stop-CM",                       10  },
 245     { "P-Coupling",                    6   },
 246     { "Calc-Ekin",                     27  },
 247     { "Lincs",                         60  },
 248     { "Lincs-Mat",                     4   },
 249     { "Shake",                         30  },
 250     { "Constraint-V",                   8  },
 251     { "Shake-Init",                    10  },
 252     { "Constraint-Vir",                24  },
 253     { "Settle",                        323 },
 254     { "Virtual Site 2",                23  },
 255     { "Virtual Site 3",                37  },
 256     { "Virtual Site 3fd",              95  },
 257     { "Virtual Site 3fad",             176 },
 258     { "Virtual Site 3out",             87  },
 259     { "Virtual Site 4fd",              110 },
 260     { "Virtual Site 4fdn",             254 },
 261     { "Virtual Site N",                 15 },
 262     { "Mixed Generalized Born stuff",   10 }
 263 };
 264
 265
 266 void init_nrnb(t_nrnb *nrnb)
 267 {
 268   int i;
 269
 270   for(i=0; (i<eNRNB); i++)
 271     nrnb->n[i]=0.0;
 272 }
 273
 274 void cp_nrnb(t_nrnb *dest, t_nrnb *src)
 275 {
 276   int i;
 277
 278   for(i=0; (i<eNRNB); i++)
 279     dest->n[i]=src->n[i];
 280 }
 281
 282 void add_nrnb(t_nrnb *dest, t_nrnb *s1, t_nrnb *s2)
 283 {
 284   int i;
 285
 286   for(i=0; (i<eNRNB); i++)
 287     dest->n[i]=s1->n[i]+s2->n[i];
 288 }
 289
 290 void print_nrnb(FILE *out, t_nrnb *nrnb)
 291 {
 292   int i;
 293
 294   for(i=0; (i<eNRNB); i++)
 295     if (nrnb->n[i] > 0)
 296       fprintf(out," %-26s %10.0f.\n",nbdata[i].name,nrnb->n[i]);
 297 }
 298
 299 void _inc_nrnb(t_nrnb *nrnb,int enr,int inc,char *file,int line)
 300 {
 301   nrnb->n[enr]+=inc;
 302 #ifdef DEBUG_NRNB
 303   printf("nrnb %15s(%2d) incremented with %8d from file %s line %d\n",
 304           nbdata[enr].name,enr,inc,file,line);
 305 #endif
 306 }
 307
 308 void print_flop(FILE *out,t_nrnb *nrnb,double *nbfs,double *mflop)
 309 {
 310   int    i;
 311   double mni,frac,tfrac,tflop;
 312   const char   *myline = "-----------------------------------------------------------------------------";
 313
 314   *nbfs = 0.0;
 315   for(i=0; (i<eNR_NBKERNEL_NR); i++) {
 316     if (strstr(nbdata[i].name,"W3-W3") != NULL)
 317       *nbfs += 9e-6*nrnb->n[i];
 318     else if (strstr(nbdata[i].name,"W3") != NULL)
 319       *nbfs += 3e-6*nrnb->n[i];
 320     else if (strstr(nbdata[i].name,"W4-W4") != NULL)
 321       *nbfs += 10e-6*nrnb->n[i];
 322     else if (strstr(nbdata[i].name,"W4") != NULL)
 323       *nbfs += 4e-6*nrnb->n[i];
 324     else
 325       *nbfs += 1e-6*nrnb->n[i];
 326   }
 327   tflop=0;
 328   for(i=0; (i<eNRNB); i++)
 329     tflop+=1e-6*nrnb->n[i]*nbdata[i].flop;
 330
 331   if (tflop == 0) {
 332     fprintf(out,"No MEGA Flopsen this time\n");
 333     return;
 334   }
 335   if (out) {
 336     fprintf(out,"\n\tM E G A - F L O P S   A C C O U N T I N G\n\n");
 337   }
 338
 339   if (out) {
 340     fprintf(out,"   RF=Reaction-Field  FE=Free Energy  SCFE=Soft-Core/Free Energy\n");
 341     fprintf(out,"   T=Tabulated        W3=SPC/TIP3p    W4=TIP4p (single or pairs)\n");
 342     fprintf(out,"   NF=No Forces\n\n");
 343
 344     fprintf(out," %-32s %16s %15s  %7s\n",
 345             "Computing:","M-Number","M-Flops","% Flops");
 346     fprintf(out,"%s\n",myline);
 347   }
 348   *mflop=0.0;
 349   tfrac=0.0;
 350   for(i=0; (i<eNRNB); i++) {
 351     mni     = 1e-6*nrnb->n[i];
 352     *mflop += mni*nbdata[i].flop;
 353     frac    = 100.0*mni*nbdata[i].flop/tflop;
 354     tfrac  += frac;
 355     if (out && mni != 0)
 356       fprintf(out," %-32s %16.6f %15.3f  %6.1f\n",
 357               nbdata[i].name,mni,mni*nbdata[i].flop,frac);
 358   }
 359   if (out) {
 360     fprintf(out,"%s\n",myline);
 361     fprintf(out," %-32s %16s %15.3f  %6.1f\n",
 362             "Total","",*mflop,tfrac);
 363     fprintf(out,"%s\n\n",myline);
 364   }
 365 }
 366
 367 void print_perf(FILE *out,double nodetime,double realtime,int nprocs,
 368                 gmx_large_int_t nsteps,real delta_t,
 369                 double nbfs,double mflop)
 370 {
 371   real runtime;
 372
 373   fprintf(out,"\n");
 374
 375   if (nodetime == 0.0) {
 376     fprintf(out,"nodetime = 0! Infinite Giga flopses!\n");
 377   }
 378 #ifdef GMX_OPENMM
 379   nodetime = realtime;
 380   fprintf(out,"\tOpenMM run - timing based on wallclock.\n\n");
 381 #else
 382   if (nprocs > 1)
 383   {
 384       nodetime = realtime;
 385       fprintf(out,"\tParallel run - timing based on wallclock.\n\n");
 386   }
 387 #endif
 388
 389   if ((nodetime > 0) && (realtime > 0)) {
 390     fprintf(out,"%12s %10s %10s %8s\n","","NODE (s)","Real (s)","(%)");
 391     fprintf(out,"%12s %10.3f %10.3f %8.1f\n","Time:",
 392             nodetime, realtime, 100.0*nodetime/realtime);
 393     if (nodetime > 60) {
 394       fprintf(out,"%12s %10s","","");
 395       pr_difftime(out,nodetime);
 396     }
 397     if (delta_t > 0) {
 398       mflop = mflop/nodetime;
 399       runtime = nsteps*delta_t;
 400       fprintf(out,"%12s %10s %10s %10s %10s\n",
 401               "","(Mnbf/s)",(mflop > 1000) ? "(GFlops)" : "(MFlops)",
 402               "(ns/day)","(hour/ns)");
 403       fprintf(out,"%12s %10.3f %10.3f %10.3f %10.3f\n","Performance:",
 404               nbfs/nodetime,(mflop > 1000) ? (mflop/1000) : mflop,
 405               runtime*24*3.6/nodetime,1000*nodetime/(3600*runtime));
 406     } else {
 407       fprintf(out,"%12s %10s %10s %14s\n",
 408               "","(Mnbf/s)",(mflop > 1000) ? "(GFlops)" : "(MFlops)",
 409               "(steps/hour)");
 410       fprintf(out,"%12s %10.3f %10.3f %14.1f\n","Performance:",
 411               nbfs/nodetime,(mflop > 1000) ? (mflop/1000) : mflop,
 412               nsteps*3600.0/nodetime);
 413     }
 414   }
 415 }
 416
 417 int cost_nrnb(int enr)
 418 {
 419   return nbdata[enr].flop;
 420 }
 421
 422 const char *nrnb_str(int enr)
 423 {
 424   return nbdata[enr].name;
 425 }
 426
 427 static const int    force_index[]={
 428   eNR_BONDS,  eNR_ANGLES,  eNR_PROPER, eNR_IMPROPER,
 429   eNR_RB,     eNR_DISRES,  eNR_ORIRES, eNR_POSRES,
 430   eNR_FBPOSRES,  eNR_NS,     eNR_NBKERNEL_OUTER
 431 };
 432 #define NFORCE_INDEX asize(force_index)
 433
 434 static const int    constr_index[]={
 435   eNR_SHAKE,     eNR_SHAKE_RIJ, eNR_SETTLE,       eNR_UPDATE,       eNR_PCOUPL,
 436   eNR_CONSTR_VIR,eNR_CONSTR_V
 437 };
 438 #define NCONSTR_INDEX asize(constr_index)
 439
 440 static double pr_av(FILE *log,t_commrec *cr,
 441                     double fav,double ftot[],const char *title)
 442 {
 443   int    i,perc;
 444   double dperc,unb;
 445
 446   unb=0;
 447   if (fav > 0) {
 448     fav /= cr->nnodes - cr->npmenodes;
 449     fprintf(log,"\n %-26s",title);
 450     for(i=0; (i<cr->nnodes); i++) {
 451         dperc=(100.0*ftot[i])/fav;
 452         unb=max(unb,dperc);
 453         perc=dperc;
 454         fprintf(log,"%3d ",perc);
 455     }
 456     if (unb > 0) {
 457       perc=10000.0/unb;
 458       fprintf(log,"%6d%%\n\n",perc);
 459     }
 460     else
 461       fprintf(log,"\n\n");
 462   }
 463   return unb;
 464 }
 465
 466 void pr_load(FILE *log,t_commrec *cr,t_nrnb nrnb[])
 467 {
 468   int    i,j,perc;
 469   double dperc,unb,uf,us;
 470   double *ftot,fav;
 471   double *stot,sav;
 472   t_nrnb *av;
 473
 474   snew(av,1);
 475   snew(ftot,cr->nnodes);
 476   snew(stot,cr->nnodes);
 477   init_nrnb(av);
 478   for(i=0; (i<cr->nnodes); i++) {
 479       add_nrnb(av,av,&(nrnb[i]));
 480       /* Cost due to forces */
 481       for(j=0; (j<eNR_NBKERNEL_NR); j++)
 482         ftot[i]+=nrnb[i].n[j]*cost_nrnb(j);
 483       for(j=0; (j<NFORCE_INDEX); j++)
 484         ftot[i]+=nrnb[i].n[force_index[j]]*cost_nrnb(force_index[j]);
 485       /* Due to shake */
 486       for(j=0; (j<NCONSTR_INDEX); j++) {
 487         stot[i]+=nrnb[i].n[constr_index[j]]*cost_nrnb(constr_index[j]);
 488       }
 489   }
 490   for(j=0; (j<eNRNB); j++)
 491     av->n[j]=av->n[j]/(double)(cr->nnodes - cr->npmenodes);
 492
 493     fprintf(log,"\nDetailed load balancing info in percentage of average\n");
 494
 495   fprintf(log," Type                 NODE:");
 496   for(i=0; (i<cr->nnodes); i++)
 497       fprintf(log,"%3d ",i);
 498   fprintf(log,"Scaling\n");
 499   fprintf(log,"---------------------------");
 500   for(i=0; (i<cr->nnodes); i++)
 501       fprintf(log,"----");
 502   fprintf(log,"-------\n");
 503
 504   for(j=0; (j<eNRNB); j++) {
 505     unb=100.0;
 506     if (av->n[j] > 0) {
 507       fprintf(log," %-26s",nrnb_str(j));
 508       for(i=0; (i<cr->nnodes); i++) {
 509           dperc=(100.0*nrnb[i].n[j])/av->n[j];
 510           unb=max(unb,dperc);
 511           perc=dperc;
 512           fprintf(log,"%3d ",perc);
 513       }
 514       if (unb > 0) {
 515         perc=10000.0/unb;
 516         fprintf(log,"%6d%%\n",perc);
 517       }
 518       else
 519         fprintf(log,"\n");
 520     }
 521   }
 522   fav=sav=0;
 523   for(i=0; (i<cr->nnodes); i++) {
 524     fav+=ftot[i];
 525     sav+=stot[i];
 526   }
 527   uf=pr_av(log,cr,fav,ftot,"Total Force");
 528   us=pr_av(log,cr,sav,stot,"Total Constr.");
 529
 530   unb=(uf*fav+us*sav)/(fav+sav);
 531   if (unb > 0) {
 532     unb=10000.0/unb;
 533     fprintf(log,"\nTotal Scaling: %.0f%% of max performance\n\n",unb);
 534   }
 535 }
 536