Skip to content
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

Implement lowering of torch.aten.exponential #2680

Merged
merged 4 commits into from
Dec 28, 2023

Conversation

godot73
Copy link
Collaborator

@godot73 godot73 commented Dec 20, 2023

#2646

Decompose aten.exponential() into: -exp(1-x)/lambda

@godot73 godot73 force-pushed the sc-exponential branch 2 times, most recently from 855b069 to 67c84b7 Compare December 21, 2023 04:51
@godot73 godot73 changed the title [WIP] Implement lowering of torch.aten.exponential Implement lowering of torch.aten.exponential Dec 21, 2023
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

@stellaraccident stellaraccident merged commit 8e389ff into llvm:main Dec 28, 2023
5 checks passed
@godot73 godot73 deleted the sc-exponential branch December 28, 2023 17:17
@godot73 godot73 restored the sc-exponential branch December 28, 2023 17:17
renxida pushed a commit to renxida/torch-mlir that referenced this pull request Dec 28, 2023
llvm#2646

Decompose aten.exponential() into: -exp(1-x)/lambda
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.

4 participants