Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion dpnp/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,14 +74,16 @@
from .dpnp_iface_utils import __all__ as _ifaceutils__all__
from ._version import get_versions
from . import exceptions as exceptions
from . import fft as fft
from . import linalg as linalg
from . import random as random
from . import scipy as scipy

__all__ = _iface__all__
__all__ += _ifaceutils__all__

# add submodules
__all__ += ["exceptions", "linalg", "scipy"]
__all__ += ["exceptions", "fft", "linalg", "random", "scipy"]


__version__ = get_versions()["version"]
Expand Down
6 changes: 2 additions & 4 deletions dpnp/dpnp_algo/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,5 @@
# THE POSSIBILITY OF SUCH DAMAGE.
# *****************************************************************************

from dpnp.dpnp_algo.dpnp_algo import *
from dpnp.dpnp_algo.dpnp_algo import __all__ as __all__dpnp_algo

__all__ = __all__dpnp_algo
# Internal package
__all__ = []
3 changes: 2 additions & 1 deletion dpnp/dpnp_array.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,8 @@
from dpctl.tensor._numpy_helper import AxisError

import dpnp
import dpnp.memory as dpm

from . import memory as dpm


def _get_unwrapped_index_key(key):
Expand Down
4 changes: 0 additions & 4 deletions dpnp/dpnp_iface.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,11 +52,7 @@
from dpctl.tensor._device import normalize_queue_device

import dpnp
from dpnp.dpnp_algo import *
from dpnp.dpnp_array import dpnp_array
from dpnp.fft import *
from dpnp.memory import *
from dpnp.random import *

__all__ = [
"are_same_logical_tensors",
Expand Down
3 changes: 2 additions & 1 deletion dpnp/dpnp_iface_bitwise.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,8 @@
import numpy

import dpnp.backend.extensions.ufunc._ufunc_impl as ufi
from dpnp.dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc

from .dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc

__all__ = [
"binary_repr",
Expand Down
3 changes: 1 addition & 2 deletions dpnp/dpnp_iface_indexing.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,8 +58,7 @@
# pylint: disable=no-name-in-module
import dpnp.backend.extensions.indexing._indexing_impl as indexing_ext

# pylint: disable=no-name-in-module
from .dpnp_algo import (
from .dpnp_algo.dpnp_algo import (
dpnp_putmask,
)
from .dpnp_array import dpnp_array
Expand Down
2 changes: 1 addition & 1 deletion dpnp/dpnp_iface_logic.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,8 +52,8 @@

import dpnp
import dpnp.backend.extensions.ufunc._ufunc_impl as ufi
from dpnp.dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc

from .dpnp_algo.dpnp_elementwise_common import DPNPBinaryFunc, DPNPUnaryFunc
from .dpnp_utils import get_usm_allocations

__all__ = [
Expand Down
2 changes: 1 addition & 1 deletion dpnp/dpnp_iface_mathematical.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@
import dpnp
import dpnp.backend.extensions.ufunc._ufunc_impl as ufi

from .dpnp_algo import dpnp_modf
from .dpnp_algo.dpnp_algo import dpnp_modf
from .dpnp_algo.dpnp_elementwise_common import (
DPNPI0,
DPNPAngle,
Expand Down
2 changes: 1 addition & 1 deletion dpnp/dpnp_iface_sorting.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@
import dpnp

# pylint: disable=no-name-in-module
from .dpnp_algo import (
from .dpnp_algo.dpnp_algo import (
dpnp_partition,
)
from .dpnp_array import dpnp_array
Expand Down
23 changes: 21 additions & 2 deletions dpnp/fft/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,26 @@

"""

from dpnp.fft.dpnp_iface_fft import *
from dpnp.fft.dpnp_iface_fft import __all__ as __all__fft
from .dpnp_iface_fft import __all__ as __all__fft
from .dpnp_iface_fft import (
fft,
fft2,
fftfreq,
fftn,
fftshift,
hfft,
ifft,
ifft2,
ifftn,
ifftshift,
ihfft,
irfft,
irfft2,
irfftn,
rfft,
rfft2,
rfftfreq,
rfftn,
)

__all__ = __all__fft
53 changes: 51 additions & 2 deletions dpnp/random/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,58 @@
# THE POSSIBILITY OF SUCH DAMAGE.
# *****************************************************************************

from .dpnp_iface_random import *
from .dpnp_iface_random import __all__ as __all__random
from .dpnp_random_state import *
from .dpnp_iface_random import (
beta,
binomial,
bytes,
chisquare,
choice,
dirichlet,
exponential,
f,
gamma,
geometric,
gumbel,
hypergeometric,
laplace,
logistic,
lognormal,
logseries,
multinomial,
multivariate_normal,
negative_binomial,
noncentral_chisquare,
noncentral_f,
normal,
pareto,
permutation,
poisson,
power,
rand,
randint,
randn,
random,
random_integers,
random_sample,
ranf,
rayleigh,
sample,
seed,
shuffle,
standard_cauchy,
standard_exponential,
standard_gamma,
standard_normal,
standard_t,
triangular,
uniform,
vonmises,
wald,
weibull,
zipf,
)
from .dpnp_random_state import RandomState
from .dpnp_random_state import __all__ as __all__random_state

__all__ = __all__random
Expand Down
1 change: 0 additions & 1 deletion dpnp/random/dpnp_iface_random.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@
import numpy

import dpnp
from dpnp.dpnp_algo import *
from dpnp.dpnp_utils import *

from .dpnp_algo_random import *
Expand Down
Loading