@@ -110,7 +110,7 @@ def _commit_descriptor(a, forward, in_place, c2c, a_strides, index, batch_fft):
110110 return dsc , out_strides
111111
112112
113- def _complex_nd_fft (
113+ def _c2c_nd_fft (
114114 a ,
115115 s ,
116116 norm ,
@@ -714,7 +714,7 @@ def dpnp_fftn(a, forward, real, s=None, axes=None, norm=None, out=None):
714714 axes = axes [- 1 ],
715715 batch_fft = a .ndim != 1 ,
716716 )
717- return _complex_nd_fft (
717+ return _c2c_nd_fft (
718718 a ,
719719 s = s [:- 1 ],
720720 norm = norm ,
@@ -728,7 +728,7 @@ def dpnp_fftn(a, forward, real, s=None, axes=None, norm=None, out=None):
728728 if c2r :
729729 # an N-D complex-to-complex FFT is performed on all axes except the
730730 # last one then a 1D complex-to-real FFT is performed on the last axis
731- a = _complex_nd_fft (
731+ a = _c2c_nd_fft (
732732 a ,
733733 s = s [:- 1 ],
734734 norm = norm ,
@@ -744,11 +744,27 @@ def dpnp_fftn(a, forward, real, s=None, axes=None, norm=None, out=None):
744744 a = _make_array_hermitian (
745745 a , axes [- 1 ], dpnp .are_same_logical_tensors (a , a_orig )
746746 )
747- return _fft (a , norm , out , forward , False , False , axes [- 1 ], a .ndim != 1 )
747+ return _fft (
748+ a ,
749+ norm = norm ,
750+ out = out ,
751+ forward = forward ,
752+ in_place = False ,
753+ c2c = False ,
754+ axes = axes [- 1 ],
755+ batch_fft = a .ndim != 1 ,
756+ )
748757
749758 # c2c
750- return _complex_nd_fft (
751- a , s , norm , out , forward , in_place , axes , a .ndim != len_axes
759+ return _c2c_nd_fft (
760+ a ,
761+ s = s ,
762+ norm = norm ,
763+ out = out ,
764+ forward = forward ,
765+ in_place = in_place ,
766+ axes = axes ,
767+ batch_fft = a .ndim != len_axes ,
752768 )
753769
754770
0 commit comments