Skip to content

Extend dpctl_ext.tensor with the remaining functions #2806

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
vlad-perevezentsev wants to merge 6 commits into
base: move_elementwise_binary_impl_part_3
Choose a base branch
from
Draft

Extend dpctl_ext.tensor with the remaining functions #2806

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
99c75cb
Use function from dpctl_ext.tensor in tensor python files
vlad-perevezentsev Mar 5, 2026
8dbf689
Move statistical functions to dpctl_ext.tensor and reuse them
vlad-perevezentsev Mar 5, 2026
e574a9a
Use _tensor_elementwise_impl from dpctl_ext
vlad-perevezentsev Mar 5, 2026
5a897c8
Add where to __init__
vlad-perevezentsev Mar 5, 2026
ecbc6a3
Move ti.allclose() to dpctl_ext.tensor
vlad-perevezentsev Mar 5, 2026
a59ccd1
Merge move_elementwise_binary_impl_part_3 into finalize_functional_mi…
vlad-perevezentsev Mar 5, 2026
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
6 changes: 6 additions & 0 deletions dpctl_ext/tensor/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,8 @@
unique_values,
)
from ._sorting import argsort, sort, top_k
from ._statistical_functions import mean, std, var
from ._testing import allclose
from ._type_utils import can_cast, finfo, iinfo, isdtype, result_type
from ._utility_functions import all, any, diff

Expand All @@ -188,6 +190,7 @@
"acosh",
"add",
"all",
"allclose",
"angle",
"any",
"arange",
Expand Down Expand Up @@ -267,6 +270,7 @@
"log10",
"max",
"maximum",
"mean",
"meshgrid",
"min",
"minimum",
Expand Down Expand Up @@ -308,6 +312,7 @@
"square",
"squeeze",
"stack",
"std",
"subtract",
"sum",
"swapaxes",
Expand All @@ -327,6 +332,7 @@
"unique_inverse",
"unique_values",
"unstack",
"var",
"vecdot",
"where",
"zeros",
Expand Down
2 changes: 1 addition & 1 deletion dpctl_ext/tensor/_clip.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,12 +28,12 @@

import dpctl
import dpctl.tensor as dpt
import dpctl.tensor._tensor_elementwise_impl as tei
from dpctl.utils import ExecutionPlacementError, SequentialOrderManager

# TODO: revert to `import dpctl.tensor...`
# when dpnp fully migrates dpctl/tensor
import dpctl_ext.tensor as dpt_ext
import dpctl_ext.tensor._tensor_elementwise_impl as tei
import dpctl_ext.tensor._tensor_impl as ti

from ._copy_utils import (
Expand Down
2 changes: 1 addition & 1 deletion dpctl_ext/tensor/_ctors.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -361,7 +361,7 @@ def _copy_through_host_walker(seq_o, usm_res):
)
is None
):
usm_res[...] = dpt.asnumpy(seq_o).copy()
usm_res[...] = dpt_ext.asnumpy(seq_o).copy()
return
else:
usm_res[...] = seq_o
Expand Down
2 changes: 1 addition & 1 deletion dpctl_ext/tensor/_reduction.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -506,7 +506,7 @@ def count_nonzero(x, /, *, axis=None, keepdims=False, out=None):
type.
"""
if x.dtype != dpt.bool:
x = dpt.astype(x, dpt.bool, copy=False)
x = dpt_ext.astype(x, dpt.bool, copy=False)
return sum(
x,
axis=axis,
Expand Down
2 changes: 1 addition & 1 deletion dpctl_ext/tensor/_set_functions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,11 +30,11 @@

import dpctl.tensor as dpt
import dpctl.utils as du
from dpctl.tensor._tensor_elementwise_impl import _not_equal, _subtract

# TODO: revert to `import dpctl.tensor...`
# when dpnp fully migrates dpctl/tensor
import dpctl_ext.tensor as dpt_ext
from dpctl_ext.tensor._tensor_elementwise_impl import _not_equal, _subtract

from ._copy_utils import _empty_like_orderK
from ._scalar_utils import (
Expand Down
Loading
Loading