Skip to content

Extend ._tensor_impl with where(), clip() and type utils functions #2778

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

Open
vlad-perevezentsev wants to merge 16 commits into
base: move_tensor_impl_accum_bool_ind_eye
Choose a base branch
from
Open

Extend ._tensor_impl with where(), clip() and type utils functions #2778

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
95ccf70
Move ti._repeat... to dpctl_ext/tensor
vlad-perevezentsev Feb 19, 2026
43f97cf
Move repeat() to dpctl_ext/tensor and reuse it in dpnp
vlad-perevezentsev Feb 19, 2026
9b248f4
Move ti._where() to dpctl_ext/tensor/libtensor
vlad-perevezentsev Feb 19, 2026
24425cf
Move _type_utils.py to dpctl_ext/tensor
vlad-perevezentsev Feb 19, 2026
c88297c
Use import _type_utils from dpctl_ext
vlad-perevezentsev Feb 20, 2026
9207fc5
Move _broadcast_shape_impl to dpctl_ext/tenspr/_manipulation_fucntion…
vlad-perevezentsev Feb 20, 2026
ebbce34
Reuse can_cast() from dpctl_ext/tensor
vlad-perevezentsev Feb 20, 2026
6a13333
Reuse finfo(), iinfo(), isdtype() from dpctl_ext/tensor
vlad-perevezentsev Feb 20, 2026
de2c429
Reuse result_type() from dpctl_ext/tensor
vlad-perevezentsev Feb 20, 2026
9a97002
Move _scalar_utils.py to dpctl_ext/tensor
vlad-perevezentsev Feb 20, 2026
15c467d
Move where() to dpctl_ext/tensor and reuse it in dpnp
vlad-perevezentsev Feb 20, 2026
5fa6ee7
Merge move_tensor_impl_accum_bool_ind_eye into move_tensor_impl_repea…
vlad-perevezentsev Feb 20, 2026
c58b531
Enable -fno-fast-math compiler flag
vlad-perevezentsev Feb 20, 2026
4f00632
Move _clip to dpctl_ext/tensor/libtensor
vlad-perevezentsev Feb 20, 2026
5d45d85
Move clip() to dpctl_ext/tensor and reuse it in dpnp
vlad-perevezentsev Feb 20, 2026
e70425b
Populate dispatch tables with where
vlad-perevezentsev Feb 20, 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
12 changes: 6 additions & 6 deletions dpctl_ext/tensor/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,10 +58,10 @@ set(_tensor_impl_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/full_ctor.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/zeros_ctor.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/triul_ctor.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/device_support_queries.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/repeat.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/repeat.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
)

set(_static_lib_trgt simplify_iteration_space)
Expand Down Expand Up @@ -92,10 +92,10 @@ endif()

set(_no_fast_math_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_and_cast_usm_to_usm.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/full_ctor.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/full_ctor.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/linear_sequences.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
# ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
)
#list(
#APPEND _no_fast_math_sources
Expand Down
14 changes: 14 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 @@ -27,6 +27,8 @@
# *****************************************************************************


from dpctl.tensor._search_functions import where

from dpctl_ext.tensor._copy_utils import (
asnumpy,
astype,
Expand All @@ -50,27 +52,39 @@
take_along_axis,
)
from dpctl_ext.tensor._manipulation_functions import (
repeat,
roll,
)
from dpctl_ext.tensor._reshape import reshape

from ._clip import clip
from ._type_utils import can_cast, finfo, iinfo, isdtype, result_type

__all__ = [
"asnumpy",
"astype",
"can_cast",
"copy",
"clip",
"extract",
"eye",
"finfo",
"from_numpy",
"full",
"iinfo",
"isdtype",
"nonzero",
"place",
"put",
"put_along_axis",
"repeat",
"reshape",
"result_type",
"roll",
"take",
"take_along_axis",
"to_numpy",
"tril",
"triu",
"where",
]
Loading
Loading