Skip to content

Commit ae1fd1f

Browse files
committed
improve code readability
1 parent 569527f commit ae1fd1f

2 files changed

Lines changed: 21 additions & 16 deletions

File tree

examples/C/collective_write.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,10 @@ pnetcdf_io(MPI_Comm comm, char *filename, int cmode, int len)
148148
bufsize *= len;
149149
}
150150

151+
if (verbose)
152+
printf("%2d: starts=%3lld %3lld %3lld count=%3lld %3lld %3lld\n",
153+
rank, starts[0],starts[1],starts[2],counts[0],counts[1],counts[2]);
154+
151155
/* allocate buffer and initialize with non-zero numbers */
152156
for (i=0; i<NUM_VARS; i++) {
153157
buf[i] = (int *) malloc(bufsize * sizeof(int));

examples/C/nonblocking_write.c

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ int main(int argc, char **argv)
126126
extern int optind;
127127
extern char *optarg;
128128
int i, j, k, err, nerrs=0, debug=0, use_contig_buf=0, use_bput=0;
129-
int nprocs, len=0, bufsize, rank;
129+
int nprocs, len=0, nelems, rank;
130130
int *sca_buf, *fix_buf[FIX_NVARS], *rec_buf[REC_NVARS];
131131
int gsizes[NDIMS], psizes[NDIMS];
132132
double write_timing, max_write_timing, write_bw;
@@ -177,10 +177,10 @@ int main(int argc, char **argv)
177177
starts[3] = rank % psizes[2];
178178

179179
counts[0] = 1;
180-
bufsize = 1;
180+
nelems = 1;
181181
for (i=0; i<NDIMS; i++) {
182182
gsizes[i] = len * psizes[i];
183-
bufsize *= len;
183+
nelems *= len;
184184
starts[i+1] *= len;
185185
counts[i+1] = len;
186186
}
@@ -193,35 +193,36 @@ int main(int argc, char **argv)
193193
/* allocate buffers */
194194
if (use_contig_buf) {
195195
/* all write buffers are allocated in a single contiguous space */
196-
size_t total_bufsize;
196+
size_t total_nelems;
197197

198-
total_bufsize = SCA_NVARS + (FIX_NVARS + REC_NVARS) * bufsize;
199-
sca_buf = (int*) malloc(total_bufsize * sizeof(int));
198+
total_nelems = SCA_NVARS + (FIX_NVARS + REC_NVARS) * nelems;
199+
200+
sca_buf = (int*) malloc(total_nelems * sizeof(int));
200201

201202
fix_buf[0] = sca_buf + SCA_NVARS;
202203
for (i=1; i<FIX_NVARS; i++)
203-
fix_buf[i] = fix_buf[i-1] + bufsize;
204+
fix_buf[i] = fix_buf[i-1] + nelems;
204205

205-
rec_buf[0] = fix_buf[FIX_NVARS-1] + bufsize;
206+
rec_buf[0] = fix_buf[FIX_NVARS-1] + nelems;
206207
for (i=1; i<REC_NVARS; i++)
207-
rec_buf[i] = rec_buf[i-1] + bufsize;
208+
rec_buf[i] = rec_buf[i-1] + nelems;
208209
}
209210
else {
210211
/* allocate individual buffers separately */
211212
sca_buf = (int*) malloc(SCA_NVARS * sizeof(int));
212213
for (i=0; i<FIX_NVARS; i++)
213-
fix_buf[i] = (int *) malloc(bufsize * sizeof(int));
214+
fix_buf[i] = (int *) malloc(nelems * sizeof(int));
214215
for (i=0; i<REC_NVARS; i++)
215-
rec_buf[i] = (int *) malloc(bufsize * sizeof(int));
216+
rec_buf[i] = (int *) malloc(nelems * sizeof(int));
216217
}
217218

218219
/* initialize buffer contents */
219220
for (j=0; j<SCA_NVARS; j++) sca_buf[j] = rank + j;
220221
for (i=0; i<FIX_NVARS; i++) {
221-
for (j=0; j<bufsize; j++) fix_buf[i][j] = rank;
222+
for (j=0; j<nelems; j++) fix_buf[i][j] = rank;
222223
}
223224
for (i=0; i<REC_NVARS; i++) {
224-
for (j=0; j<bufsize; j++) rec_buf[i][j] = rank;
225+
for (j=0; j<nelems; j++) rec_buf[i][j] = rank;
225226
}
226227

227228
MPI_Barrier(MPI_COMM_WORLD);
@@ -329,7 +330,7 @@ int main(int argc, char **argv)
329330
else { /* write using bput */
330331

331332
/* bbufsize must be max of data type converted before and after */
332-
bbufsize = (SCA_NVARS + bufsize * (FIX_NVARS + REC_NVARS)) * sizeof(int);
333+
bbufsize = (SCA_NVARS + nelems * (FIX_NVARS + REC_NVARS)) * sizeof(int);
333334
err = ncmpi_buffer_attach(ncid, bbufsize);
334335
ERR
335336

@@ -393,7 +394,7 @@ int main(int argc, char **argv)
393394

394395
write_timing = MPI_Wtime() - write_timing;
395396

396-
write_size = bufsize * (FIX_NVARS + NTIMES * REC_NVARS) + SCA_NVARS;
397+
write_size = nelems * (FIX_NVARS + NTIMES * REC_NVARS) + SCA_NVARS;
397398
write_size *= sizeof(int);
398399

399400
MPI_Reduce(&write_size, &sum_write_size, 1, MPI_OFFSET, MPI_SUM, 0, MPI_COMM_WORLD);
@@ -416,7 +417,7 @@ int main(int argc, char **argv)
416417
printf("Total amount writes (include header) = %lld bytes\n", sum_write_size);
417418
printf("Total amount writes reported by pnetcdf (include header) = %lld bytes\n", sum_put_size);
418419
printf("\n");
419-
float subarray_size = (float)bufsize*sizeof(int)/1048576.0;
420+
float subarray_size = (float)nelems*sizeof(int)/1048576.0;
420421
print_info(&info_used);
421422
printf("Local array size %d x %d x %d integers, size = %.2f MB\n",len,len,len,subarray_size);
422423
sum_write_size /= 1048576.0;

0 commit comments

Comments
 (0)