Skip to content

add tensor linalg extension #2799

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
ndgrigorian wants to merge 6 commits into
base: move_tensor_elementwise_impl_unary_par_3
Choose a base branch
from
Open

add tensor linalg extension #2799

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
3333fc1
add tensor linalg extension
ndgrigorian Mar 5, 2026
fc34f30
Merge move_tensor_elementwise_impl_unary_par_3 into move-tensor-linalg
vlad-perevezentsev Mar 5, 2026
3bd1b0e
fix includes and namespaces in dot.cpp
ndgrigorian Mar 5, 2026
1620f77
use dpctl_ext.tensor throughout _linear_algebra_functions
ndgrigorian Mar 5, 2026
75580a5
fix clang-format
ndgrigorian Mar 6, 2026
7e6a283
abbreviate namespaces in linalg module sources and headers
ndgrigorian Mar 6, 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
16 changes: 15 additions & 1 deletion dpctl_ext/tensor/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,10 @@ set(_sorting_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/sorting/searchsorted.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/sorting/topk.cpp
)
set(_linalg_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/elementwise_functions/elementwise_functions_type_utils.cpp
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/linalg_functions/dot.cpp
)
set(_tensor_accumulation_impl_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/tensor_accumulation.cpp
${_accumulator_sources}
Expand All @@ -182,6 +186,10 @@ set(_tensor_sorting_impl_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/tensor_sorting.cpp
${_sorting_sources}
)
set(_tensor_linalg_impl_sources
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/tensor_linalg.cpp
${_linalg_sources}
)

set(_static_lib_trgt simplify_iteration_space)

Expand Down Expand Up @@ -228,6 +236,12 @@ add_sycl_to_target(TARGET ${python_module_name} SOURCES ${_tensor_sorting_impl_s
target_link_libraries(${python_module_name} PRIVATE ${_static_lib_trgt})
list(APPEND _py_trgts ${python_module_name})

set(python_module_name _tensor_linalg_impl)
pybind11_add_module(${python_module_name} MODULE ${_tensor_linalg_impl_sources})
add_sycl_to_target(TARGET ${python_module_name} SOURCES ${_tensor_linalg_impl_sources})
target_link_libraries(${python_module_name} PRIVATE ${_static_lib_trgt})
list(APPEND _py_trgts ${python_module_name})

set(_clang_prefix "")
if(WIN32)
set(_clang_prefix "/clang:")
Expand All @@ -245,7 +259,7 @@ list(
${_elementwise_sources}
${_reduction_sources}
${_sorting_sources}
# ${_linalg_sources}
${_linalg_sources}
${_accumulator_sources}
)

Expand Down
10 changes: 10 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 @@ -107,6 +107,12 @@
take,
take_along_axis,
)
from ._linear_algebra_functions import (
matmul,
matrix_transpose,
tensordot,
vecdot,
)
from ._manipulation_functions import (
broadcast_arrays,
broadcast_to,
Expand Down Expand Up @@ -216,6 +222,8 @@
"min",
"moveaxis",
"permute_dims",
"matmul",
"matrix_transpose",
"negative",
"nonzero",
"ones",
Expand Down Expand Up @@ -251,6 +259,7 @@
"take_along_axis",
"tan",
"tanh",
"tensordot",
"tile",
"top_k",
"to_numpy",
Expand All @@ -262,6 +271,7 @@
"unique_inverse",
"unique_values",
"unstack",
"vecdot",
"where",
"zeros",
"zeros_like",
Expand Down
Loading
Loading