-
Notifications
You must be signed in to change notification settings - Fork 21
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
feat: move transposes up (and then down) #1099
Conversation
forgot to add broadcastindim is reshape to td, though I will merge it with the canon today anyways |
d409a35
to
2297685
Compare
i might have enabled conflicting optimizations |
632e1ca
to
7518034
Compare
ad689d8
to
169b182
Compare
failures due to EnzymeAD/Enzyme-JAX#586 |
55d215b
to
38e713d
Compare
This is not working in GB-25: PRONTOLab/GB-25#128 (comment) |
Let me try and patch it |
and I broke more stuff... |
Also @wsmoses one of the 5 commented out reshape passes are probably broken https://github.com/EnzymeAD/Reactant.jl/actions/runs/14195378107/job/39769233267?pr=1099#step:10:576 |
actually need to check which one is broken, that is not the one I activated in the last commit |
Sp the problem in the GB repo is not fixed yet. The thing which ends up happening is that the input buffers can now be aliased and we need to be careful while donating it |
dd8a43b
to
1e01733
Compare
feat: try collapsing the ops again fix: syntax feat: bump jll fix: print for debugging fix: unary transpose fix: try fixing fix: conflicting passes fix: conflicting passes feat: newer opts fix: reshape pad fix: avoid transpose_reduce_window for now Update to Reactant_jll v0.0.116 fix: add some aliasing checks feat: enable more passes fix: disable some passes again fix: types fix: rollback some changes fix: rollback more passes feat: store if an arg is donated fix: mark donated fix: remove broadcast_in_dim_is_reshape for now fix: reshape_pad is also broken fix: concat is busted not pad
No description provided.