Skip to content

Commit 77ba49c

Browse files
committed
Fixing slab results in docs to reflect new default behaviour
1 parent 0dccb25 commit 77ba49c

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

docs/source/global.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ We note that a low-level approach to creating such a distributed array would
101101
be::
102102

103103
import numpy as np
104-
from mpi4py_fft import Pencil, Subcomm
104+
from mpi4py_fft.pencil import Pencil, Subcomm
105105
from mpi4py import MPI
106106
comm = MPI.COMM_WORLD
107107
N = (8, 8)

docs/source/parallel.rst

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -99,14 +99,16 @@ and axis 1 will be transformed first, such that the global output array
9999
will be of shape (128, 65, 128). The distributed input and output arrays
100100
will now have shape::
101101

102-
0 (128, 128, 64)
103-
1 (128, 128, 64)
102+
0 (64, 128, 128)
103+
1 (64, 128, 128)
104104

105-
0 (64, 65, 128)
106-
1 (64, 65, 128)
105+
0 (128, 33, 128)
106+
1 (128, 32, 128)
107107

108-
Note that the input array will be distributed in axis 2 and the
109-
output in axis 0.
108+
Note that the input array will still be distributed in axis 0 and the
109+
output in axis 1. This order of distribution can be tweaked using the
110+
`grid` keyword. Setting `grid=(1, 1, -1)` will force the last axis
111+
to be distributed on the input array.
110112

111113
Another way to tweak the distribution is to use the :class:`.Subcomm`
112114
class directly::

0 commit comments

Comments
 (0)