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

Add capturable ASGD impl #107857

Closed
wants to merge 25 commits into from
Closed

Add capturable ASGD impl #107857

wants to merge 25 commits into from

Commits on Jul 31, 2023

  1. Add capturable param to ASGD

    @mlazos
    mlazos committed Jul 31, 2023
    Configuration menu
    Copy the full SHA
    dfbccdd View commit details
    Browse the repository at this point in the history

Commits on Aug 24, 2023

  1. Merge branch 'main' into mlazos/asgd-compile

    @mlazos
    mlazos committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    2b1a31b View commit details
    Browse the repository at this point in the history

  2. Add capturable impl for asgd

    @mlazos
    mlazos committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    6e4f357 View commit details
    Browse the repository at this point in the history

  3. Fix lint

    @mlazos
    mlazos committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    d6340be View commit details
    Browse the repository at this point in the history

  4. Remove comment

    @mlazos
    mlazos committed Aug 24, 2023
    Configuration menu
    Copy the full SHA
    e76c283 View commit details
    Browse the repository at this point in the history

Commits on Aug 30, 2023

  1. Merge branch 'main' into mlazos/asgd-compile

    @mlazos
    mlazos committed Aug 30, 2023
    Configuration menu
    Copy the full SHA
    7f553b0 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into mlazos/asgd-compile

    @mlazos
    mlazos committed Aug 30, 2023
    Configuration menu
    Copy the full SHA
    9f7cb11 View commit details
    Browse the repository at this point in the history

  3. Fix kernel counts

    @mlazos
    mlazos committed Aug 30, 2023
    Configuration menu
    Copy the full SHA
    bb3e30e View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2023

  1. Start minimizing intermediates

    @mlazos
    mlazos committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    67a36ac View commit details
    Browse the repository at this point in the history

  2. Stage 2

    @mlazos
    mlazos committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    9979309 View commit details
    Browse the repository at this point in the history

  3. Better capturable impl

    @mlazos
    mlazos committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    08a2dda View commit details
    Browse the repository at this point in the history

  4. Merge branch 'main' into mlazos/asgd-compile

    @mlazos
    mlazos committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    66b61e4 View commit details
    Browse the repository at this point in the history

  5. Make cudagraphable by default

    @mlazos
    mlazos committed Aug 31, 2023
    Configuration menu
    Copy the full SHA
    b9337ac View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2023

  1. Small fixes

    @mlazos
    mlazos committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    dce1dd4 View commit details
    Browse the repository at this point in the history

  2. Working with minimal kernels

    @mlazos
    mlazos committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    6a9f583 View commit details
    Browse the repository at this point in the history

  3. Working solution

    @mlazos
    mlazos committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    3b5e26b View commit details
    Browse the repository at this point in the history

  4. Don't use param data type

    @mlazos
    mlazos committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    d496ab1 View commit details
    Browse the repository at this point in the history

  5. Add cudagraph tests and capturable for asgd

    @mlazos
    mlazos committed Sep 1, 2023
    Configuration menu
    Copy the full SHA
    4870e0b View commit details
    Browse the repository at this point in the history

Commits on Sep 6, 2023

  1. Merge branch 'main' into mlazos/asgd-compile

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    8649a36 View commit details
    Browse the repository at this point in the history

  2. Undo foreac_lerp meta registration

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    32873a4 View commit details
    Browse the repository at this point in the history

  3. Fix peak mem test with capturable error

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    808c705 View commit details
    Browse the repository at this point in the history

  4. Fix doc

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    abf7013 View commit details
    Browse the repository at this point in the history

  5. Add test with non foreach and foreach for agsgd

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    08ab847 View commit details
    Browse the repository at this point in the history

  6. Add missing kwargs

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    2467ef3 View commit details
    Browse the repository at this point in the history

  7. Fix nit

    @mlazos
    mlazos committed Sep 6, 2023
    Configuration menu
    Copy the full SHA
    d072d9d View commit details
    Browse the repository at this point in the history