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.

Sign up for GitHub

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

Jump to bottom

Free CuArrays in the reverse pass #1340

Closed
wants to merge 7 commits into from
Closed

Free CuArrays in the reverse pass #1340

wants to merge 7 commits into from

Commits on Dec 18, 2022

  1. add only_once flag to Context, for pullback non-re-use

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    125e8a2 View commit details
    Browse the repository at this point in the history

  2. define maybe_final, insert into a few adjoint rules (map, broadcast)

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    424df8f View commit details
    Browse the repository at this point in the history

  3. insert maybe_final into all CR rrules

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    2524163 View commit details
    Browse the repository at this point in the history

  4. insert maybe_final into all ZygoteRules at-adjoint rules

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    a1a6896 View commit details
    Browse the repository at this point in the history

  5. use adjoint_final from ZygoteRules, broadcasting should opt-in

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    759ca60 View commit details
    Browse the repository at this point in the history

  6. skip kron test on 1.9+

    @mcabbott
    mcabbott committed Dec 18, 2022
    Configuration menu
    Copy the full SHA
    7e6e31e View commit details
    Browse the repository at this point in the history

Commits on Dec 19, 2022

  1. fixup

    @mcabbott
    mcabbott committed Dec 19, 2022
    Configuration menu
    Copy the full SHA
    9f01eff View commit details
    Browse the repository at this point in the history