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

feat(train) Add support for torch.compile (EXPERIMENTAL) #2931

Open
wants to merge 27 commits into
base: main
Choose a base branch
from
Open

feat(train) Add support for torch.compile (EXPERIMENTAL) #2931

wants to merge 27 commits into from

Commits on Aug 5, 2024

  1. Prototype compile

    @canergen
    canergen committed Aug 5, 2024
    Configuration menu
    Copy the full SHA
    e58c2c3 View commit details
    Browse the repository at this point in the history

Commits on Aug 7, 2024

  1. Merge branch 'main' into can_compile

    @canergen
    canergen committed Aug 7, 2024
    Configuration menu
    Copy the full SHA
    568ea81 View commit details
    Browse the repository at this point in the history

  2. Changes for compiled

    @canergen
    canergen committed Aug 7, 2024
    Configuration menu
    Copy the full SHA
    7e207b9 View commit details
    Browse the repository at this point in the history

  3. Add full support for compile

    @canergen
    canergen committed Aug 7, 2024
    Configuration menu
    Copy the full SHA
    a25b7e4 View commit details
    Browse the repository at this point in the history

  4. [pre-commit.ci] auto fixes from pre-commit.com hooks 

    for more information, see https://pre-commit.ci
    @pre-commit-ci
    pre-commit-ci[bot] committed Aug 7, 2024
    Configuration menu
    Copy the full SHA
    7301f30 View commit details
    Browse the repository at this point in the history

Commits on Aug 26, 2024

  1. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Aug 26, 2024
    Configuration menu
    Copy the full SHA
    a0e2aaa View commit details
    Browse the repository at this point in the history

  2. fixed poissonVi GPU support, 

    trainingplans with future imports and test_compute_kl revert
    @ori-kron-wis
    ori-kron-wis committed Aug 26, 2024
    Configuration menu
    Copy the full SHA
    0cd6a2c View commit details
    Browse the repository at this point in the history

Commits on Aug 28, 2024

  1. updated linux cuda test yml

    @ori-kron-wis
    ori-kron-wis committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    8d7df18 View commit details
    Browse the repository at this point in the history

  2. updated linux cuda test yml

    @ori-kron-wis
    ori-kron-wis committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    50ea265 View commit details
    Browse the repository at this point in the history

  3. updated linux cuda test yml

    @ori-kron-wis
    ori-kron-wis committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    c17ac7f View commit details
    Browse the repository at this point in the history

  4. updated linux cuda test yml to the current branch runtime image

    @ori-kron-wis
    ori-kron-wis committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    437f1ae View commit details
    Browse the repository at this point in the history

  5. updated linux cuda test yml to the current branch dev image

    @ori-kron-wis
    ori-kron-wis committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    91420c1 View commit details
    Browse the repository at this point in the history

Commits on Aug 29, 2024

  1. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Aug 29, 2024
    Configuration menu
    Copy the full SHA
    f5309da View commit details
    Browse the repository at this point in the history

  2. add torch compile tests to most models (expect pyro and jax)

    @ori-kron-wis
    ori-kron-wis committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    41a6317 View commit details
    Browse the repository at this point in the history

Commits on Sep 15, 2024

  1. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Sep 15, 2024
    Configuration menu
    Copy the full SHA
    a80b402 View commit details
    Browse the repository at this point in the history

Commits on Sep 16, 2024

  1. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Sep 16, 2024
    Configuration menu
    Copy the full SHA
    0148292 View commit details
    Browse the repository at this point in the history

  2. Added Optional to fix import which was related to previous py3.9 fix

    @ori-kron-wis
    ori-kron-wis committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    dafffb0 View commit details
    Browse the repository at this point in the history

Commits on Sep 17, 2024

  1. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Sep 17, 2024
    Configuration menu
    Copy the full SHA
    4246022 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Sep 17, 2024
    Configuration menu
    Copy the full SHA
    5292652 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into can_compile

    @ori-kron-wis
    ori-kron-wis authored Sep 17, 2024
    Configuration menu
    Copy the full SHA
    572db62 View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2024

  1. Merge remote-tracking branch 'origin/main' into can_compile

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    6a778bb View commit details
    Browse the repository at this point in the history

  2. update main, fix jax tests to pass

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    372f870 View commit details
    Browse the repository at this point in the history

  3. update cuda on docker?

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    71beae5 View commit details
    Browse the repository at this point in the history

  4. adding jax[cuda] and nccl-cu12 to linux cuda test installation

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    66a11bc View commit details
    Browse the repository at this point in the history

  5. Merge remote-tracking branch 'origin/main' into can_compile

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    d459c71 View commit details
    Browse the repository at this point in the history

  6. tried w/o pyro test

    @ori-kron-wis
    ori-kron-wis committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    e3673e8 View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2024

  1. added a flag for test_pyto_regression optional so rest of CUDA tests … 

    …could pass
    @ori-kron-wis
    ori-kron-wis committed Sep 19, 2024
    Configuration menu
    Copy the full SHA
    33453ca View commit details
    Browse the repository at this point in the history