add tensor linalg extension by ndgrigorian · Pull Request #2799 · IntelPython/dpnp

ndgrigorian · 2026-03-04T22:10:03Z

This PR migrates the _tensor_linalg_impl extension to dpctl_ext.tensor and extends dpctl_ext.tensor Python API with dpctl.tensor functions matmul, matrix_transpose, tensordot, and vecdot

Have you provided a meaningful PR description?
Have you added a test, reproducer or referred to an issue with a reproducer?
Have you tested your changes locally for CPU and GPU devices?
Have you made sure that new changes do not introduce compiler warnings?
Have you checked performance impact of proposed changes?
Have you added documentation for your changes, if necessary?
Have you added your changes to the changelog?

dpctl_ext/tensor/libtensor/source/linalg_functions/dot.cpp

dpctl_ext/tensor/libtensor/source/linalg_functions/dot_dispatch.hpp

…_impl_unary_par_2

…ntwise_impl_unary_par_3

vlad-perevezentsev

LGTM
Thank you @ndgrigorian for your help with the tensor migration!

dpctl_ext/tensor/libtensor/source/linalg_functions/dot_atomic_support.hpp

dpctl_ext/tensor/libtensor/include/kernels/linalg_functions/dot_product.hpp

dpctl_ext/tensor/libtensor/include/kernels/linalg_functions/gemm.hpp

github-actions · 2026-03-18T13:15:22Z

Array API standard conformance tests for dpnp=0.20.0dev3=py313h509198e_225 ran successfully.
Passed: 1356
Failed: 1
Skipped: 16

vlad-perevezentsev added 30 commits February 3, 2026 12:16

Rename folder dpctl to dpctl_ext

d4fd805

Add simplify_iteration_space implementation to libtensor

c040713

Extend codespell ignore list for libtensor

14b466f

Add copy_and_cast kernels to libtensor

dcc421b

Add copy_usm_ndarray_into_usm_ndarray implementation

5a9c14c

Add pybind11 bindings for dpctl_ext.tensor._tensor_impl

4f63340

Add CMake build files for dpctl_ext

634579c

Add empty __init__ to dpctl_ext/

79d40f2

Enable _same_logical_tensors in _tensor_impl

7949c17

Add device_support_queries to enable default device types

29d6c02

Enable building and packaging of dpctl_ext

936e719

Use _tensor_impl from dpctl_ext.tensor in dpnp

cd85f1e

Move put() and take() to dpctl_ext/tensor

0c6780a

Use put/take from dpctl_ext.tensor in dpnp

87e5482

Move full() to dpctl_ext/tensor

b537f30

Use full and _full_usm_ndarray from dpctl_ext in dpnp

d50f263

Update .gitignore to ignore .so files in dpctl_ext

f189dc5

Move _zeros_usm_ndarray to dpctl_ext

f9a1817

Use _zeros_usm_ndarray from dpctl_ext in dpnp_fill.py

4b8505a

Move linear-sequence implementations to dpctl_ext/tensor

61106b2

Use _tensor_impl from dpctl_ext in dpnp_utils_fft.py

a030579

Move tril()/triu() to dpctl_ext/tensor

a1d6fa3

Use tril/triu/_tril from dpctl_ext.tensor in dpnp

f1d6e56

Disable pylint no-name-in-module for dpctl_ext

6680790

Add TODO comments

263b717

Use default_device_complex_type from dpctl_ext on test_array_api_info.py

4130c1b

Remove unused build_dpctl_ext function

17ca9ab

Apply remarks for CMake files

79cb2a4

Apply remarks for c++ files

4bf080e

Remove linear-sequence implementations

cfa6cd6

vlad-perevezentsev reviewed Mar 11, 2026

View reviewed changes

dpctl_ext/tensor/libtensor/source/linalg_functions/dot.cpp Show resolved Hide resolved

vlad-perevezentsev added 2 commits March 11, 2026 08:00

Add the missing includes for dot.cpp

c0ff29f

Small clean-up bitwise_invert.hpp

d5bbfc0

antonwolfy reviewed Mar 12, 2026

View reviewed changes

address review feedback

7fbe3e9

ndgrigorian requested review from antonwolfy and vlad-perevezentsev March 12, 2026 14:54

vlad-perevezentsev added 12 commits March 13, 2026 04:37

Apply remarks

d80a7dc

Merge move_tensor_elementwise_impl_unary into move_tensor_elementwise…

2d5b148

…_impl_unary_par_2

Apply remarks

9ce7e7f

Merge move_tensor_elementwise_impl_unary_par_2 into move_tensor_eleme…

4b7e2f8

…ntwise_impl_unary_par_3

Apply remarks

5fbac89

Merge include-dpctl-tensor into move_tensor_elementwise_impl_unary_par_2

d5d0e36

Update a wrong function name in docs

ffee731

Merge move_tensor_elementwise_impl_unary_par_2 into move_tensor_eleme…

927471e

…ntwise_impl_unary_par_3

Fix typos in docs

0e3bc76

Merge include-dpctl-tensor into move_tensor_elementwise_impl_unary_par_3

75cf7ee

Increase timeout-minutes for the public CI

011a393

Merge move_tensor_elementwise_impl_unary_par_3 into move-tensor-linalg

8238d95

Base automatically changed from move_tensor_elementwise_impl_unary_par_3 to include-dpctl-tensor March 18, 2026 11:26

vlad-perevezentsev added 2 commits March 18, 2026 04:26

Merge include-dpctl-tensor into move-tensor-linalg

31f7455

Merge move-tensor-linalg into move-tensor-linalg

412328f

vlad-perevezentsev previously approved these changes Mar 18, 2026

View reviewed changes

antonwolfy reviewed Mar 18, 2026

View reviewed changes

Apply remarks

026ed93

vlad-perevezentsev dismissed their stale review via 026ed93 March 18, 2026 12:05

antonwolfy approved these changes Mar 18, 2026

View reviewed changes

vlad-perevezentsev merged commit b0647db into include-dpctl-tensor Mar 18, 2026
17 of 33 checks passed

vlad-perevezentsev deleted the move-tensor-linalg branch March 18, 2026 12:25

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

add tensor linalg extension#2799

add tensor linalg extension#2799
vlad-perevezentsev merged 204 commits intoinclude-dpctl-tensorfrom
move-tensor-linalg

ndgrigorian commented Mar 4, 2026

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

vlad-perevezentsev left a comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

github-actions bot commented Mar 18, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Conversation

ndgrigorian commented Mar 4, 2026

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

vlad-perevezentsev left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

github-actions bot commented Mar 18, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants