Skip to content

[TORCH] Add support for logcumsumexp Op #4187

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Jun 9, 2025

Conversation

sharavana20
Copy link
Contributor

@sharavana20 sharavana20 commented May 19, 2025

  • Decomposed logcumsumexp op into Aten ops.
    • The decomposition follows the formula:
      logcumsumexp(x) = log(cumsum(exp(x)))
  • Added test cases in the e2e part.

This implementation addresses and closes #4183

@sharavana20 sharavana20 force-pushed the logcumsumexp branch 4 times, most recently from cdba452 to 6a20e5e Compare May 20, 2025 11:18
@sharavana20 sharavana20 marked this pull request as ready for review May 20, 2025 11:31
@sharavana20
Copy link
Contributor Author

sharavana20 commented May 20, 2025

Hi @vivekkhandelwal1 @penguin-wwy @AmosLewis @zjgarvey — I’d be grateful if any of you could take a look at this PR. Your feedback would be greatly appreciated!

@sharavana20
Copy link
Contributor Author

sharavana20 commented May 27, 2025

@vivekkhandelwal1 Thanks a lot for the feedback. I’ve updated the code based on your comments.

Copy link
Collaborator

@vivekkhandelwal1 vivekkhandelwal1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@sharavak
Copy link

sharavak commented Jun 9, 2025

Thanks for the review, @vivekkhandelwal1.
I’ve resolved the conflicts — would you mind merging it when you get a chance?

@vivekkhandelwal1 vivekkhandelwal1 enabled auto-merge (squash) June 9, 2025 16:46
@vivekkhandelwal1 vivekkhandelwal1 merged commit 1ac131c into llvm:main Jun 9, 2025
3 checks passed
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.

[TORCH] Add support for logcumsumexp
3 participants