Skip to content

Commit 42306ce

Browse files
author
Vahid Tavanashad
committed
improve formatting
1 parent 83f375a commit 42306ce

File tree

1 file changed

+22
-6
lines changed

1 file changed

+22
-6
lines changed

dpnp/fft/dpnp_utils_fft.py

Lines changed: 22 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)