Skip to content

Extend ._tensor_impl with advanced indexing functions#2777

Draft
vlad-perevezentsev wants to merge 9 commits intomove_tensor_impl_copyfrom
move_tensor_impl_accum_bool_ind_eye
Draft

Extend ._tensor_impl with advanced indexing functions#2777
vlad-perevezentsev wants to merge 9 commits intomove_tensor_impl_copyfrom
move_tensor_impl_accum_bool_ind_eye

Conversation

@vlad-perevezentsev
Copy link
Contributor

This PR extends _tensor_impl in dpctl_ext.tensor with theadvanced indexing (_extract, _place, _nonzero, mask_positions, ), repeat (_cumsum_1d) and _eye functions

It also adds eye(), extract(), nonzero(), place(), put_along_axis(), take_along_axis() to dpctl_ext.tensor and updates the corresponding dpnp functions to use these implementations internally

  • 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?

@github-actions
Copy link
Contributor

Array API standard conformance tests for dpnp=0.20.0dev3=py313h509198e_70 ran successfully.
Passed: 1355
Failed: 2
Skipped: 7

@github-actions
Copy link
Contributor

View rendered docs @ https://intelpython.github.io/dpnp/pull/2777/index.html

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant

Comments