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

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

sharavana20
Copy link

@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 3 times, most recently from 7c66a84 to cdba452 Compare May 20, 2025 10:37
@sharavana20 sharavana20 marked this pull request as ready for review May 20, 2025 11:31
@sharavana20
Copy link
Author

@vivekkhandelwal1 I don’t have permission to assign reviewers directly, but could you please take a look at this PR when you have a moment? Your review would be much appreciated!

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
1 participant