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

Remove special handling of multiple inputs to pullbacks from ChainRules #1

Merged
merged 6 commits into from
May 20, 2020

Conversation

oxinabox
Copy link
Owner

@oxinabox oxinabox commented May 7, 2020

This is a PR against FluxML#366
However, it needs a bunch of other stuff to be tagged first.
So it should not be treated as blocking FluxML#366 as that might be a while.
I have tested it locally,
it does give a bit of a performance boost.
(Not huge but a bit)

Mostly it makes the code cleaner.

@oxinabox
Copy link
Owner Author

oxinabox commented May 7, 2020

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>
@oxinabox oxinabox merged commit bf913a2 into oxinabox:ox/chainrules_step1b May 20, 2020
oxinabox added a commit that referenced this pull request May 21, 2020
…es (#1)

* ChainRules pullbacks always have 1 input JuliaDiff/ChainRulesCore.jl#152

* swap to version of chainrules that don't use multiarg pullbacks

* update tests

* make so don't need custom rule anymore

* add comment

* Update src/compiler/chainrules.jl

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>
oxinabox added a commit that referenced this pull request May 21, 2020
…es (#1)

* ChainRules pullbacks always have 1 input JuliaDiff/ChainRulesCore.jl#152

* swap to version of chainrules that don't use multiarg pullbacks

* update tests

* make so don't need custom rule anymore

* add comment

* Update src/compiler/chainrules.jl

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>
oxinabox added a commit that referenced this pull request May 21, 2020
…es (#1)

* ChainRules pullbacks always have 1 input JuliaDiff/ChainRulesCore.jl#152

* swap to version of chainrules that don't use multiarg pullbacks

* update tests

* make so don't need custom rule anymore

* add comment

* Update src/compiler/chainrules.jl

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>
@oxinabox oxinabox deleted the ox/cr_1pb branch May 21, 2020 13:53
oxinabox added a commit that referenced this pull request May 28, 2020
…es (#1)

* ChainRules pullbacks always have 1 input JuliaDiff/ChainRulesCore.jl#152

* swap to version of chainrules that don't use multiarg pullbacks

* update tests

* make so don't need custom rule anymore

* add comment

* Update src/compiler/chainrules.jl

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>

Co-authored-by: willtebbutt <wt0881@my.bristol.ac.uk>
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