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 vmap support for torch.nn.functional.smooth_l1_loss #98357

Closed
wants to merge 7 commits into from
Closed

Add vmap support for torch.nn.functional.smooth_l1_loss #98357

wants to merge 7 commits into from

Commits on Apr 4, 2023

  1. Added vmap support for torch.nn.functional.smooth_l1_loss

    @yhl48
    yhl48 committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    0c43d23 View commit details
    Browse the repository at this point in the history

  2. Removed xfail

    @yhl48
    yhl48 committed Apr 4, 2023
    Configuration menu
    Copy the full SHA
    b7817ea View commit details
    Browse the repository at this point in the history

Commits on Apr 13, 2023

  1. Added a helper function loss_batch_rule_helper

    @yhl48
    yhl48 committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    3b1865f View commit details
    Browse the repository at this point in the history

  2. Revert "Removed xfail" 

    This reverts commit b7817ea.
    @yhl48
    yhl48 committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    20bea84 View commit details
    Browse the repository at this point in the history

  3. Removed xfail for nn.functional.smooth_l1_loss

    @yhl48
    yhl48 committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    a8d3e09 View commit details
    Browse the repository at this point in the history

Commits on Apr 14, 2023

  1. Use template instead of std::function

    @yhl48
    yhl48 committed Apr 14, 2023
    Configuration menu
    Copy the full SHA
    b6a93fd View commit details
    Browse the repository at this point in the history

  2. Cleaned up code and renamed arguments

    @yhl48
    yhl48 committed Apr 14, 2023
    Configuration menu
    Copy the full SHA
    03570dd View commit details
    Browse the repository at this point in the history