Skip to content

[bug] Fix decomposition logic for logaddexp#4509

Open
vihangm wants to merge 1 commit intollvm:mainfrom
gimletlabs:vihang/logaddexp_broadcast
Open

[bug] Fix decomposition logic for logaddexp#4509
vihangm wants to merge 1 commit intollvm:mainfrom
gimletlabs:vihang/logaddexp_broadcast

Conversation

@vihangm
Copy link
Copy Markdown

@vihangm vihangm commented Mar 22, 2026

The result types on the exp ops during decomposition were incorrect. The AddTensorOp is where the auto broadcasting happens so the exp ops should propagate the input type as their output type.

I added a couple of tests that exhibit the issue, and fail without this patch but pass after.

This was broken from the original implementation in #4201

Signed-off-by: Vihang Mehta vihang@gimletlabs.ai

Signed-off-by: Vihang Mehta <vihang@gimletlabs.ai>
@vihangm vihangm force-pushed the vihang/logaddexp_broadcast branch from 044617b to b912b25 Compare March 23, 2026 18:29
@vihangm
Copy link
Copy Markdown
Author

vihangm commented Mar 31, 2026

@penguin-wwy or @vivekkhandelwal1 let me know if you'd be able to review this.

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.

1 participant