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:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/functional/common.cpp 3 additions, 46 deletionsoneflow/api/python/functional/common.cpp
- oneflow/api/python/functional/common.h 12 additions, 6 deletionsoneflow/api/python/functional/common.h
- oneflow/api/python/functional/indexing.cpp 197 additions, 0 deletionsoneflow/api/python/functional/indexing.cpp
- oneflow/api/python/functional/indexing.h 44 additions, 0 deletionsoneflow/api/python/functional/indexing.h
- oneflow/api/python/functional/python_arg.cpp 80 additions, 23 deletionsoneflow/api/python/functional/python_arg.cpp
- oneflow/core/cuda/atomic.cuh 5 additions, 0 deletionsoneflow/core/cuda/atomic.cuh
- oneflow/core/functional/functional_api.yaml 4 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/array_functor.cpp 30 additions, 8 deletionsoneflow/core/functional/impl/array_functor.cpp
- oneflow/core/functional/tensor_index.cpp 188 additions, 17 deletionsoneflow/core/functional/tensor_index.cpp
- oneflow/core/functional/tensor_index.h 13 additions, 3 deletionsoneflow/core/functional/tensor_index.h
- oneflow/user/kernels/expand_kernel.cpp 2 additions, 0 deletionsoneflow/user/kernels/expand_kernel.cpp
- oneflow/user/kernels/expand_kernel.cu 2 additions, 0 deletionsoneflow/user/kernels/expand_kernel.cu
- python/oneflow/framework/functional.py 3 additions, 0 deletionspython/oneflow/framework/functional.py
- python/oneflow/nn/modules/gather_nd.py 1 addition, 8 deletionspython/oneflow/nn/modules/gather_nd.py
- python/oneflow/test/tensor/test_tensor_indexing.py 163 additions, 0 deletionspython/oneflow/test/tensor/test_tensor_indexing.py
Please register or sign in to comment