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 Pytorch memory stats along with PyNVML #81

Merged
merged 7 commits into from
Oct 31, 2023
Merged

Add Pytorch memory stats along with PyNVML #81

merged 7 commits into from
Oct 31, 2023

Commits on Oct 31, 2023

  1. added max memory allocated and reserved

    @IlyasMoutawwakil
    IlyasMoutawwakil committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    777aa35 View commit details
    Browse the repository at this point in the history

  2. updated llama results with mem stats

    @IlyasMoutawwakil
    IlyasMoutawwakil committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    6a25a6d View commit details
    Browse the repository at this point in the history

  3. fix indexes

    @IlyasMoutawwakil
    IlyasMoutawwakil committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    d650c0d View commit details
    Browse the repository at this point in the history

  4. add cache emptiying between forward and generate

    @IlyasMoutawwakil
    IlyasMoutawwakil committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    e31cd4c View commit details
    Browse the repository at this point in the history

  5. remove unnecessary resets

    @actions-user
    actions-user committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    f2b3f8a View commit details
    Browse the repository at this point in the history

  6. update reporting tool to target a memory type

    @actions-user
    actions-user committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    96ae22a View commit details
    Browse the repository at this point in the history

  7. deprecate peak memory

    @IlyasMoutawwakil
    IlyasMoutawwakil committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    620ae7f View commit details
    Browse the repository at this point in the history