Skip to content

Commit 02cdaec

Browse files
committed
Explicilty cast chunk_shape to int32 to fix numpy 2 int promotion changes
1 parent 4735501 commit 02cdaec

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

dedalus/core/transposes.pyx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ cdef class FFTWTranspose:
4141
cdef readonly int datasize, axis
4242
cdef readonly int N0, N1, N2, N3
4343
cdef readonly int[::1] global_shape
44+
cdef readonly int[::1] chunk_shape
4445
cdef readonly int[::1] col_starts
4546
cdef readonly int[::1] row_starts
4647
cdef readonly int[::1] col_ends
@@ -60,6 +61,7 @@ cdef class FFTWTranspose:
6061
logger.debug("Building FFTW transpose plan for (dtype, gshape, axis) = (%s, %s, %s)" %(dtype, global_shape, axis))
6162
# Attributes
6263
self.global_shape = global_shape = np.array(global_shape, dtype=np.int32)
64+
self.chunk_shape = chunk_shape = np.array(chunk_shape, dtype=np.int32)
6365
self.datasize = {np.float64: 1, np.complex128: 2}[np.dtype(dtype).type]
6466
self.axis = axis
6567
self.pycomm = pycomm

0 commit comments

Comments
 (0)