For large batch sizes, the multiplication of the number of grid points
with the number of states in a batch could overflow. Change the
integers involved to 8-byte.