Skip to content

vectorize-pr#856

Open
ipdemes wants to merge 88 commits intonv-legate:branch-24.03from
ipdemes:vectorize
Open

vectorize-pr#856
ipdemes wants to merge 88 commits intonv-legate:branch-24.03from
ipdemes:vectorize

Conversation

@ipdemes
Copy link
Copy Markdown
Contributor

@ipdemes ipdemes commented Mar 24, 2023

This PR adds support for np.vectorize to cuNumeric
It depends on nv-legate/legate#640

This implementation has following limitations:


  1. It requires all output arrays or scalars be the same type and shape. This is something that we will probably have to require, at least I don’t see the way to support this in a performant way

  2. It requres all input arrays be the same shape. This is something we can fix with broadcasting.

  3. There is currently no support for “excluded” and “signature”



I had to manually add typings for Numba to avoid pre-commit errors. I don’t think I did it right and asked @bryevdv to look at it. 







And thank you @bryevdv for helping with fixing documentation issues

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

Labels

category:new-feature PR introduces a new feature and will be classified as such in release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants