BladeDISC icon indicating copy to clipboard operation
BladeDISC copied to clipboard

Published 20 hours ago • verified publisher icon alibaba

[WIP]Add AutoOffloadingPass

Open Yancey1989 opened this issue 7 months ago • 0 comments

this PR added a offloading pass to offloading a large live range buffer to the host and reloading to device at the right place to reduce the memory peak. benchmark on part training graph, this pass reduced 32% peak memory: image

Yancey1989 avatar Jun 28 '24 05:06 Yancey1989