Skip to content

Argwhere and flatnonzero#525

Merged
mfoerste4 merged 10 commits intonv-legate:branch-22.10from
mfoerste4:argwhere_flatnonzero
Aug 24, 2022
Merged

Argwhere and flatnonzero#525
mfoerste4 merged 10 commits intonv-legate:branch-22.10from
mfoerste4:argwhere_flatnonzero

Conversation

@mfoerste4
Copy link
Contributor

This PR adds functionality for np.argwhere and np.flatnonzero to cuNumeric.

np.argwhere

Implemented similar to np.nonzero task, but with different return shape

np.flatnonzero

Just forwarded to np.nonzero(np.ravel(a))[0]

@mfoerste4 mfoerste4 requested a review from magnatelee August 17, 2022 09:26
@mfoerste4
Copy link
Contributor Author

@magnatelee I suggested you for review, feel free to forward to someone else.

@magnatelee
Copy link
Contributor

left two petty comments. otherwise looks good to me. great to see that you reuse the kernels from the nonzero task!

@mfoerste4 mfoerste4 requested a review from magnatelee August 22, 2022 08:51
@mfoerste4
Copy link
Contributor Author

left two petty comments. otherwise looks good to me. great to see that you reuse the kernels from the nonzero task!
@magnatelee , please verify requested changes to unblock merge.

@magnatelee
Copy link
Contributor

@mfoerste4 approved. feel free to merge it whenever you can.

@mfoerste4 mfoerste4 merged commit 07ac802 into nv-legate:branch-22.10 Aug 24, 2022
@mfoerste4 mfoerste4 deleted the argwhere_flatnonzero branch August 24, 2022 17:01
jjwilke pushed a commit to jjwilke/cunumeric that referenced this pull request Sep 2, 2022
* extract counting routine to re-use with argwhere

* added argwhere and flatnonzero

* combine tests

* add doc entry

* fix mypy issues

* adjusted test_config expectations

* cleanup
manopapad pushed a commit that referenced this pull request Feb 18, 2025
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.

2 participants