Skip to content

Support batched pinv#4686

Merged
toslunar merged 6 commits intocupy:masterfrom
leofang:batched_pinv
Mar 2, 2021
Merged

Support batched pinv#4686
toslunar merged 6 commits intocupy:masterfrom
leofang:batched_pinv

Conversation

@leofang
Copy link
Copy Markdown
Member

@leofang leofang commented Feb 19, 2021

Blocked by #4684 as batched SVD is required but currently emitting random errors in the CI.

Close #3062.

  • For supporting batched pinv: the original implementation already followed NumPy's, so really it's just adding newaxis here and there to broadcast.
  • This PR also fixes an incompatibility when the input array is of zero size

@kmaehashi kmaehashi added cat:feature New features/APIs prio:medium st:blocked-by-another-pr Blocked by another pull-request labels Feb 22, 2021
@leofang leofang changed the title [WIP] Support batched pinv Support batched pinv Feb 22, 2021
@leofang leofang marked this pull request as ready for review February 22, 2021 08:14
@leofang
Copy link
Copy Markdown
Member Author

leofang commented Feb 22, 2021

Jenkins, test this please

@chainer-ci
Copy link
Copy Markdown
Member

Jenkins CI test (for commit 902c11d, target branch master) succeeded!

@leofang
Copy link
Copy Markdown
Member Author

leofang commented Feb 25, 2021

Thanks, @toslunar! Your suggestions make it much cleaner 👍

@leofang
Copy link
Copy Markdown
Member Author

leofang commented Feb 26, 2021

Jenkins, test this please

@chainer-ci
Copy link
Copy Markdown
Member

Jenkins CI test (for commit da610ee, target branch master) succeeded!

@toslunar toslunar added this to the v9.0.0rc1 milestone Mar 2, 2021
@toslunar toslunar removed the st:blocked-by-another-pr Blocked by another pull-request label Mar 2, 2021
@toslunar toslunar merged commit 57ef68c into cupy:master Mar 2, 2021
@leofang leofang deleted the batched_pinv branch March 2, 2021 06:18
@leofang
Copy link
Copy Markdown
Member Author

leofang commented Mar 2, 2021

Thanks, @toslunar!

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

cupy.linalg.{svd, pinv} should support broadcasting

4 participants