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

Fix dense elements attribute in Enzyme.autodiff #593 #604

Merged
merged 2 commits into from
Jan 25, 2025

Conversation

mofeing
Copy link
Collaborator

@mofeing mofeing commented Jan 24, 2025

Fixes #593

(I still have a problem with rev-rule that i wasn't having before but that's probably on my side)

Copy link
Member

@wsmoses wsmoses left a comment

Choose a reason for hiding this comment

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

Add a test?

@mofeing
Copy link
Collaborator Author

mofeing commented Jan 24, 2025

It looks like I uncovered another bug? This seems to fix my issue in Tenet

@wsmoses
Copy link
Member

wsmoses commented Jan 24, 2025

Seems so, can you extract the unoptimized mlir, and make an issue on EnzymeJaX with the required passes for it to err?

@mofeing
Copy link
Collaborator Author

mofeing commented Jan 25, 2025

Fixed in EnzymeAD/Enzyme-JAX#272

Just need to update the commit, build and retest

@mofeing mofeing mentioned this pull request Jan 25, 2025
@wsmoses wsmoses force-pushed the ss/fix/denselementsattr-autodiff branch from 94160cd to bc0a683 Compare January 25, 2025 23:02
@wsmoses wsmoses merged commit 67575fb into main Jan 25, 2025
36 of 39 checks passed
@wsmoses wsmoses deleted the ss/fix/denselementsattr-autodiff branch January 25, 2025 23:38
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.

Tenet + Reactant + Enzyme.gradient broken on last releases
2 participants