IntelPython / dpnp

Data Parallel Extension for NumPy
BSD 2-Clause "Simplified" License
97 stars 22 forks source link

Update `dpnp.extract` implementation to get rid of limitations for input arguments #1906

Closed antonwolfy closed 3 months ago

antonwolfy commented 3 months ago

The PR proposes to extend implementation of dpnp.extract and to align the behavior with NumPy.

Docstrings are updated with missing information about input arguments. The tests are updated to cover more use cases and to include compute follows data scenario.

Additionally, the implementation of dpnp.extract is now properly handling async execution model from dpctl.

github-actions[bot] commented 3 months ago

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