mirror/netcdf-c
2
0
Fork 0
mirror of https://github.com/Unidata/netcdf-c.git synced 2024-12-09 08:11:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

working on 2D decomposition code

Browse Source
This commit is contained in:
Edward Hartnett 2020-07-04 10:45:33 -06:00
parent 7a9397aed8
commit 86c39cf7a2
1 changed files with 27 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
nc_perf/tst_gfs_data_1.c
View File

@ -230,8 +230,34 @@ write_metadata(int ncid, int *data_varid, int s, int f, int deflate, int *dim_le
/* Based on the MPI rank and number of tasks, calculate the
* decomposition of the 2D lat/lon coordinate variables. */
int
decomp_2D(int my_rank, int mpi_size, int *dim_len, size_t *start, size_t *count)
decomp_2D(int my_rank, int mpi_size, int *dim_len, size_t *latlon_start,
size_t *latlon_count)
{
/* Size of local arrays (i.e. for this pe) lon and lat data. This
* is specific to 4 pes. */
latlon_count[0] = dim_len[0]/2;
if (my_rank == 0 || my_rank == 2)
{
latlon_start[0] = 0;
}
else
{
latlon_start[0] = dim_len[0]/2;
}
latlon_count[1] = dim_len[1]/2;
if (my_rank == 0 || my_rank == 1)
{
latlon_start[1] = 0;
}
else
{
latlon_start[1] = dim_len[1]/2;
}
printf("%d: latlon_start %ld %ld latlon_count %ld %ld\n", my_rank, latlon_start[0],
latlon_start[1], latlon_count[0], latlon_count[1]);
return 0;
}
@ -346,27 +372,6 @@ main(int argc, char **argv)
if (my_rank == mpi_size - 1)
phalf_loc_size = phalf_loc_size + dim_len[3] % mpi_size;
/* Size of local arrays (i.e. for this pe) lon and lat data. This is */
/* specific to 4 pes. */
latlon_count[0] = GRID_YT_LEN;
if (my_rank == 0 || my_rank == 2)
{
latlon_start[0] = 0;
}
else
{
latlon_start[0] = GRID_YT_LEN;
}
latlon_count[1] = 768;
if (my_rank == 0 || my_rank == 1)
{
latlon_start[1] = 0;
}
else
{
latlon_start[1] = 768;
}
/* Allocate space on this pe to hold the coordinate var data for this pe. */
if (!(value_pfull_loc = malloc(data_count[1] * sizeof(float)))) ERR;
if (!(value_phalf_loc = malloc(phalf_loc_size * sizeof(float)))) ERR;