Skip to content

GitLab

Switch branch/tag
Find file HistoryPermalink
  • Quentin Duval's avatar
    [feat] layer memory tracking (#808) · ad92220c
    Quentin Duval authored 
    

* [feat] layer memory tracking

* [feat] layer memory tracking (add tests in CI)

* [feat] layer memory tracking: doc typos

* [feat] layer memory tracking: mypy fixes

* [feat] layer memory tracking: fixes for FSDP all gather tracking on pytorch 1.9 and above

* [feat] layer memory tracking: lint

* [feat] layer memory tracking: mypy
Co-authored-by: default avatarQuentinDuval <QuentinDuval@users.noreply.github.com>
    ad92220c
__init__.py 0 Bytes