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

Adding integration test for FSDP Memory Tracking and Estimation #426

Merged
merged 2 commits into from
Jun 25, 2024
Merged

Adding integration test for FSDP Memory Tracking and Estimation #426

merged 2 commits into from
Jun 25, 2024

Commits on Jun 25, 2024

  1. Adding integration test for FSDP Memory Tracking and Estimation 

    [ghstack-poisoned]
    @sanketpurandare
    sanketpurandare committed Jun 25, 2024
    Configuration menu
    Copy the full SHA
    3c6a391 View commit details
    Browse the repository at this point in the history

  2. Update on "Adding integration test for FSDP Memory Tracking and Estim… 

    …ation"


Adds an integration test for `FSDPMemTracker` which will help keep `estimation.py` in sync with `train.py`.

`python test_runner.py test_outputs  --test fsdp2_mem_tracker`

cc: gnadathur 

[ghstack-poisoned]
    @sanketpurandare
    sanketpurandare committed Jun 25, 2024
    Configuration menu
    Copy the full SHA
    79ddab2 View commit details
    Browse the repository at this point in the history