Skip to content

Commit 49b9342

Browse files
Add functions to copy array into C-/F- contig destination
This is done more efficiently than generic copy-and-cast kernel. It is also done yet more efficiently for the batch of square matrices. Copy from (batch of views into C-contig matrices) to F-contig array of the same shape. src.shape = (n, n, ....) src.strides = (ld_src, 1, ...) Copy from (batch of views into F-contig matrices) to C-contig array of the same shape src.shape = (..., n, n) src.strides = (..., 1, ld_src)
1 parent 3f6af83 commit 49b9342

File tree

6 files changed

+1388
-1
lines changed

6 files changed

+1388
-1
lines changed

dpctl/tensor/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ set(_tensor_impl_sources
123123
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/tensor_ctors.cpp
124124
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/accumulators.cpp
125125
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_and_cast_usm_to_usm.cpp
126+
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_as_contig.cpp
126127
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_numpy_ndarray_into_usm_ndarray.cpp
127128
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_for_reshape.cpp
128129
${CMAKE_CURRENT_SOURCE_DIR}/libtensor/source/copy_for_roll.cpp

0 commit comments

Comments
 (0)