Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Paul Johnson's avatar
    [FSDP] ssd_offload fixing backward path (grad_fn) for SsdFlatParameter and... · 51b53ddb
    Paul Johnson authored 
    [FSDP] ssd_offload fixing backward path (grad_fn) for SsdFlatParameter and SsdFlatParameterView (#974)

* [FSDP] fixing backward path for SsdFlatParameter and SsdFlatParameterView when overriding .data

* Get ssd_offload unit tests passing

* [FSDP] get all test_fsdp_offload tests passing w/ ssd_offload on

* Update changelog
    51b53ddb
test_fsdp_offload.py 20.2 KB