Skip to content

Commit 2d2bfc2

Browse files
Initial check-in of elementwise functions: abs/cos/isnan/add
1 parent 1320d39 commit 2d2bfc2

File tree

16 files changed

+3772
-0
lines changed

16 files changed

+3772
-0
lines changed

dpctl/tensor/CMakeLists.txt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,15 @@ pybind11_add_module(${python_module_name} MODULE
4545
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/triul_ctor.cpp
4646
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
4747
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/device_support_queries.cpp
48+
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
4849
)
50+
set(_clang_prefix "")
51+
if (WIN32)
52+
set(_clang_prefix "/clang:")
53+
endif()
54+
set_source_files_properties(
55+
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
56+
PROPERTIES COMPILE_OPTIONS "${_clang_prefx}-fno-approx-func")
4957
target_compile_options(${python_module_name} PRIVATE -fno-sycl-id-queries-fit-in-int)
5058
target_link_options(${python_module_name} PRIVATE -fsycl-device-code-split=per_kernel)
5159
if(UNIX)

dpctl/tensor/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@
9090
from dpctl.tensor._usmarray import usm_ndarray
9191

9292
from ._constants import e, inf, nan, newaxis, pi
93+
from ._elementwise_funcs import abs, add, cos, isnan
9394

9495
__all__ = [
9596
"Device",
@@ -164,4 +165,8 @@
164165
"pi",
165166
"nan",
166167
"inf",
168+
"abs",
169+
"add",
170+
"cos",
171+
"isnan",
167172
]

0 commit comments

Comments
 (0)