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

[PyTorch] Branching operations #1027

Merged
merged 15 commits into from
Aug 10, 2024
Merged

[PyTorch] Branching operations #1027

merged 15 commits into from
Aug 10, 2024

Commits on Jul 17, 2024

  1. Add op for in-place add 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 17, 2024
    Configuration menu
    Copy the full SHA
    9f61bca View commit details
    Browse the repository at this point in the history

  2. Add op for in-place add 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 17, 2024
    Configuration menu
    Copy the full SHA
    2cda094 View commit details
    Browse the repository at this point in the history

  3. Add op that adds extra output to fuser 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 17, 2024
    Configuration menu
    Copy the full SHA
    6bf2869 View commit details
    Browse the repository at this point in the history

Commits on Jul 18, 2024

  1. Add fused op for GEMM+bias+add 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 18, 2024
    Configuration menu
    Copy the full SHA
    6d78177 View commit details
    Browse the repository at this point in the history

  2. Add fused op for dgrad+add 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 18, 2024
    Configuration menu
    Copy the full SHA
    da7a981 View commit details
    Browse the repository at this point in the history

Commits on Jul 19, 2024

  1. Add documentation 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 19, 2024
    Configuration menu
    Copy the full SHA
    872f863 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into branching-ops

    @timmoon10
    timmoon10 committed Jul 19, 2024
    Configuration menu
    Copy the full SHA
    93c4040 View commit details
    Browse the repository at this point in the history

  3. [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 Jul 19, 2024
    Configuration menu
    Copy the full SHA
    151df0b View commit details
    Browse the repository at this point in the history

  4. Fix linter warnings 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Jul 19, 2024
    Configuration menu
    Copy the full SHA
    4e618cd View commit details
    Browse the repository at this point in the history

Commits on Jul 22, 2024

  1. Merge branch 'main' into branching-ops

    @timmoon10
    timmoon10 authored Jul 22, 2024
    Configuration menu
    Copy the full SHA
    fadbb8a View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2024

  1. Merge branch 'main' into branching-ops

    @timmoon10
    timmoon10 committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    8b78f65 View commit details
    Browse the repository at this point in the history

Commits on Aug 3, 2024

  1. Review suggestions from @ptrendx 

    Output tensor dtype and device take precedence over weight tensor in linear functional API. Move some index calculation to fuser constructor. Avoid some unnecessary dereferences.

Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Aug 3, 2024
    Configuration menu
    Copy the full SHA
    2679fbf View commit details
    Browse the repository at this point in the history

Commits on Aug 5, 2024

  1. Debug test failures 

    Signed-off-by: Tim Moon <tmoon@nvidia.com>
    @timmoon10
    timmoon10 committed Aug 5, 2024
    Configuration menu
    Copy the full SHA
    33347f4 View commit details
    Browse the repository at this point in the history

Commits on Aug 9, 2024

  1. Update transformer_engine/pytorch/ops/fuser.py 

    Signed-off-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>
    @timmoon10
    timmoon10 authored Aug 9, 2024
    Configuration menu
    Copy the full SHA
    a5418ab View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into branching-ops

    @timmoon10
    timmoon10 authored Aug 9, 2024
    Configuration menu
    Copy the full SHA
    912742b View commit details
    Browse the repository at this point in the history