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

make torch compile OSS compatible #803

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

make torch compile OSS compatible #803

wants to merge 4 commits into from

Commits on Apr 25, 2024

  1. enfore pytorch >= 2.0 

    Differential Revision: D56446353
    @JKSenthil @facebook-github-bot
    JKSenthil authored and facebook-github-bot committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    4594bf0 View commit details
    Browse the repository at this point in the history

  2. add and use generic torch version comparator 

    Differential Revision: D56446382
    @JKSenthil @facebook-github-bot
    JKSenthil authored and facebook-github-bot committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    5e51dd5 View commit details
    Browse the repository at this point in the history

  3. replace _init_optim_state w/ tnt's util 

    Differential Revision:
D56446429

Split of "[tnt] test against stable pytorch version"
    @JKSenthil @facebook-github-bot
    JKSenthil authored and facebook-github-bot committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    7a3a2f4 View commit details
    Browse the repository at this point in the history

  4. make torch compile callback OSS compatible 

    Summary: Touches few torch.compile calls in TorchTNT which rely on nightlies to be OSS friendly

Reviewed By: galrotem

Differential Revision: D56483268
    @JKSenthil @facebook-github-bot
    JKSenthil authored and facebook-github-bot committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    380aa23 View commit details
    Browse the repository at this point in the history