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

[torch.compile] register allreduce operations as custom ops #8526

Merged
merged 13 commits into from
Sep 17, 2024
Merged

[torch.compile] register allreduce operations as custom ops #8526

merged 13 commits into from
Sep 17, 2024

Commits on Sep 13, 2024

  1. init version

    @youkaichao
    youkaichao committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    a1f1931 View commit details
    Browse the repository at this point in the history

Commits on Sep 16, 2024

  1. Merge branch 'main' into register_allreduce2

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    7c8fccb View commit details
    Browse the repository at this point in the history

  2. remove spaces

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    bb2615b View commit details
    Browse the repository at this point in the history

  3. move should_custom_ar to python

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    bbdfd6b View commit details
    Browse the repository at this point in the history

  4. remove should_custom_ar from c

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    9b4b802 View commit details
    Browse the repository at this point in the history

  5. slight changes

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    8376f35 View commit details
    Browse the repository at this point in the history

  6. add tests

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    82e9581 View commit details
    Browse the repository at this point in the history

  7. add tests

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    d25c8a4 View commit details
    Browse the repository at this point in the history

  8. type annotation

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    244bbd4 View commit details
    Browse the repository at this point in the history

  9. fix type

    @youkaichao
    youkaichao committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    2da8f06 View commit details
    Browse the repository at this point in the history

Commits on Sep 17, 2024

  1. add one more condition

    @youkaichao
    youkaichao committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    7da3dc8 View commit details
    Browse the repository at this point in the history

  2. update tests

    @youkaichao
    youkaichao committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    9f9359c View commit details
    Browse the repository at this point in the history

  3. add package

    @youkaichao
    youkaichao committed Sep 17, 2024
    Configuration menu
    Copy the full SHA
    e32ee56 View commit details
    Browse the repository at this point in the history