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

Support batching in MLIR autodiff operations #2172

Closed
wants to merge 4 commits into from

Conversation

jumerckx
Copy link
Collaborator

RE: #2170
Still very much in progress.

@jumerckx
Copy link
Collaborator Author

jumerckx commented Nov 20, 2024

I added the width to autodiff as well, now trying to hook things up to actually work seems to be a bit more involved 😅.
d075b82 (#2172) contains some debugging code, I think all the shadowtypes need to be adapted for width != 1, but then I'm unsure how to continue.

EDIT: removed this WIP commit, proper batched code generation is left for a follow-up pr.

@jumerckx jumerckx marked this pull request as ready for review November 21, 2024 09:47
@wsmoses
Copy link
Member

wsmoses commented Nov 21, 2024

@jumerckx can you fix the format?

@jumerckx
Copy link
Collaborator Author

jumerckx commented Nov 22, 2024

image
This PR is bugged, GitHub has been saying this for over a day now.
Reopened this here #2173

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.

2 participants