Skip to content

Commit 4eb83aa

Browse files
committed
Implements dpctl.tensor.clip
1 parent 1d57614 commit 4eb83aa

File tree

8 files changed

+1997
-0
lines changed

8 files changed

+1997
-0
lines changed

dpctl/tensor/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ set(_tensor_impl_sources
5050
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
5151
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/repeat.cpp
5252
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/reduction_over_axis.cpp
53+
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
5354
)
5455

5556
set(python_module_name _tensor_impl)
@@ -65,6 +66,7 @@ set(_no_fast_math_sources
6566
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/linear_sequences.cpp
6667
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
6768
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/reduction_over_axis.cpp
69+
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/clip.cpp
6870
)
6971
foreach(_src_fn ${_no_fast_math_sources})
7072
get_source_file_property(_cmpl_options_prop ${_src_fn} COMPILE_OPTIONS)

dpctl/tensor/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@
9393
from dpctl.tensor._usmarray import usm_ndarray
9494
from dpctl.tensor._utility_functions import all, any
9595

96+
from ._clip import clip
9697
from ._constants import e, inf, nan, newaxis, pi
9798
from ._elementwise_funcs import (
9899
abs,
@@ -322,4 +323,5 @@
322323
"exp2",
323324
"copysign",
324325
"rsqrt",
326+
"clip",
325327
]

0 commit comments

Comments
 (0)