Skip to content
Snippets Groups Projects
Unverified Commit db71898c authored by Houjiang Chen's avatar Houjiang Chen Committed by GitHub
Browse files

Dev advanced indexing (#5559)


* Support numpy advanced indexing.

* Finally adjust if continuos subspace.

* Finish integer array indexing and combining indexing.

* Finish index array parsing.

* Add tensor indexing unittest

* Add check

* Refine

* Refine tensor getitem unittest

* Support int64_t cuda atomicAdd.

* Not use PyObjectPtr since PySequence_Fast_GET_ITEM returns borrowed reference.

* Remove experimental in indexing unittest.

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent f73a2533
No related branches found
No related tags found
No related merge requests found
Showing
with 747 additions and 111 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment