benchmarks/experimental/offload.py · c19cc89726359b458357d983057a305ddfb343ba · OpenDAS / fairscale

[offload] Add support for record_function when using OffloadModel (#564) · c19cc897

anj-s authored Apr 01, 2021

* add record_function support

* add more record_function cutpoints

* add more record_function cutpoints

* lint errors

* make string ids more specific

c19cc897

offload.py 15.8 KB

Replace offload.py