diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/LoadAware.c | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/src/LoadAware.c b/src/LoadAware.c index ea01efd..fcff53d 100644 --- a/src/LoadAware.c +++ b/src/LoadAware.c @@ -166,7 +166,7 @@ int PUGH_SetPartitionInfo (int dim, const char *partition_info[]) int PUGH_GetBounds (int dim, int *bounds[], int nprocs[], int nsize[]) { DECLARE_CCTK_PARAMETERS - int retval, manual, external_manual; + int i, retval, manual, external_manual; const char *partition_info; @@ -208,8 +208,18 @@ int PUGH_GetBounds (int dim, int *bounds[], int nprocs[], int nsize[]) } else { - CCTK_WARN (1, "Only 1D, 2D, and 3D supported"); - retval = -1; + if (manual) + { + CCTK_WARN (1, "Only 1D, 2D, and 3D supported in manual topology setup"); + retval = -1; + } + else + { + for (i = 0; i < dim; i++) + { + PUGH_GetSliceSizes (nprocs[i], nsize[i], NULL, &bounds[i], 0); + } + } } return (retval); |