Merge pull request #1203 from IntelPython/feature/elementwise-functions

Initial check-in of some unary and binary elementwise functions

Author: oleksandr-pavlyk

.github/workflows/conda-package.yml

@@ -44,7 +44,7 @@ jobs:
         run: conda install conda-build
       - name: Build conda package
         run: |
-          CHANNELS="-c intel -c main --override-channels"
+          CHANNELS="-c dppy/label/tools -c intel -c main --override-channels"
           VERSIONS="--python ${{ matrix.python }}"
           TEST="--no-test"
           conda build \
@@ -161,13 +161,20 @@ jobs:
         run: |
           . $CONDA/etc/profile.d/conda.sh
           conda activate test_dpctl
-          python -c "import dpctl; dpctl.lsplatform()"
+          python -c "import dpctl; dpctl.lsplatform(verbosity=2)"
+      - name: Install gdb
+        run: |
+          sudo apt-get install -y gdb
+      - name: Run test_elementwise under gdb
+        run: |
+          . $CONDA/etc/profile.d/conda.sh
+          conda activate test_dpctl
+          gdb --batch -ex r -ex 'info sharedlibrary' -ex 'set print elements 1000' -ex bt --args ${CONDA_PREFIX}/bin/python -m pytest -q -ra --disable-warnings --pyargs dpctl.tests.test_tensor_elementwise::test_cos_order -vv || true
       - name: Run tests
         run: |
           . $CONDA/etc/profile.d/conda.sh
           conda activate test_dpctl
-          # clinfo -l
-          python -m pytest --pyargs $MODULE_NAME
+          python -m pytest -v --pyargs $MODULE_NAME
 
   test_windows:
     needs: build_windows

conda-recipe/meta.yaml

@@ -14,7 +14,7 @@ requirements:
     build:
         - {{ compiler('cxx') }}
         - {{ compiler('dpcpp') }} >=2023.1  # [not osx]
-        - sysroot_linux-64 >=2.17  # [linux]
+        - sysroot_linux-64 >=2.28  # [linux]
     host:
         - setuptools
         - cmake  >=3.21

conda-recipe/run_test.sh

@@ -3,5 +3,5 @@
 set -e
 
 ${PYTHON} -c "import dpctl; print(dpctl.__version__)"
-${PYTHON} -c "import dpctl; dpctl.lsplatform()"
+${PYTHON} -c "import dpctl; dpctl.lsplatform(verbosity=2)"
 ${PYTHON} -m pytest -q -ra --disable-warnings -p no:faulthandler --cov dpctl --cov-report term-missing --pyargs dpctl -vv

dpctl/tensor/CMakeLists.txt

@@ -46,7 +46,15 @@ pybind11_add_module(${python_module_name} MODULE
     ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/where.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/boolean_reductions.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/device_support_queries.cpp
+    ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
 )
+set(_clang_prefix "")
+if (WIN32)
+  set(_clang_prefix "/clang:")
+endif()
+set_source_files_properties(
+  ${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions.cpp
+  PROPERTIES COMPILE_OPTIONS "${_clang_prefix}-fno-fast-math")
 target_compile_options(${python_module_name} PRIVATE -fno-sycl-id-queries-fit-in-int)
 target_link_options(${python_module_name} PRIVATE -fsycl-device-code-split=per_kernel)
 if(UNIX)

dpctl/tensor/__init__.py

@@ -91,6 +91,17 @@
 from dpctl.tensor._utility_functions import all, any
 
 from ._constants import e, inf, nan, newaxis, pi
+from ._elementwise_funcs import (
+    abs,
+    add,
+    cos,
+    divide,
+    equal,
+    isfinite,
+    isinf,
+    isnan,
+    sqrt,
+)
 
 __all__ = [
     "Device",
@@ -167,4 +178,13 @@
     "pi",
     "nan",
     "inf",
+    "abs",
+    "add",
+    "cos",
+    "isinf",
+    "isnan",
+    "isfinite",
+    "sqrt",
+    "divide",
+    "equal",
 ]