So if I do `context.empty((1, 2, 3))`, then `ndistdim == 1` so the `grid_shape = (comm_size,)`. But then `len(grid_shape) != ndim` as you say above.