Skip to content

Commit 1364af6

Browse files
authored
move fft to desc partially (#757)
1 parent 408ffed commit 1364af6

File tree

2 files changed

+73
-86
lines changed

2 files changed

+73
-86
lines changed

dpnp/fft/dpnp_algo_fft.pyx

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,8 @@ and the rest of the library
3333
"""
3434

3535

36-
import dpnp
3736
from dpnp.dpnp_algo cimport *
38-
from dpnp.dpnp_utils cimport *
37+
cimport dpnp.dpnp_utils as utils
3938

4039

4140
__all__ = [
@@ -45,12 +44,12 @@ __all__ = [
4544
ctypedef void(*fptr_dpnp_fft_fft_t)(void *, void * , long * , long * , size_t, long, long, size_t)
4645

4746

48-
cpdef dparray dpnp_fft(dparray input, size_t input_boundarie, size_t output_boundarie, long axis, size_t inverse):
47+
cpdef dparray dpnp_fft(utils.dpnp_descriptor input, size_t input_boundarie, size_t output_boundarie, long axis, size_t inverse):
4948

5049
cdef dparray_shape_type input_shape = input.shape
5150
cdef dparray_shape_type output_shape = input_shape
5251

53-
cdef long axis_norm = normalize_axis((axis,), input_shape.size())[0]
52+
cdef long axis_norm = utils.normalize_axis((axis,), input_shape.size())[0]
5453
output_shape[axis_norm] = output_boundarie
5554

5655
# convert string type names (dparray.dtype) to C enum DPNPFuncType

0 commit comments

Comments
 (0)