Skip to content

Commit 8f75dec

Browse files
rfftn_numpy and irfftn_numpy should use iterative fft/ifft when operating of 2D arrays
1 parent d71aa89 commit 8f75dec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mkl_fft/_pydfti.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -818,7 +818,7 @@ def rfftn_numpy(x, s=None, axes=None):
818818
ss = list(s)
819819
ss[-1] = a.shape[la]
820820
a = _fix_dimensions(a, tuple(ss), axes)
821-
if len(set(axes)) == len(axes) and len(axes) == a.ndim:
821+
if len(set(axes)) == len(axes) and len(axes) == a.ndim and len(axes) > 2:
822822
ss, aa = _remove_axis(s, axes, la)
823823
ind = [slice(None,None,1),] * len(s)
824824
for ii in range(a.shape[la]):
@@ -842,7 +842,7 @@ def irfftn_numpy(x, s=None, axes=None):
842842
if not no_trim:
843843
a = _fix_dimensions(a, s, axes)
844844
ovr_x = True if _datacopied(<cnp.ndarray> a, x) else False
845-
if len(set(axes)) == len(axes) and len(axes) == a.ndim:
845+
if len(set(axes)) == len(axes) and len(axes) == a.ndim and len(axes) > 2:
846846
ss, aa = _remove_axis(s, axes, la)
847847
ind = [slice(None,None,1),] * len(s)
848848
for ii in range(a.shape[la]):

0 commit comments

Comments
 (0)