mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
80527c0cf2
Summary: This adds function to model container doing weight swapping with double buffering. There are 2 parts for double buffering a) Write constants into inactive buffer b) Swap active buffer For (a), we write the constants into the buffer that's currently not in use, and store the information in both constants map and the corresponding constant array to read. For (b), we obtain the lock, and activate the constant map/constant array that is inactive, and flag the one that's currently in use to inactive. Test Plan: test/cpp/aot_inductor/test.cpp Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D51543732](https://our.internmc.facebook.com/intern/diff/D51543732) Pull Request resolved: https://github.com/pytorch/pytorch/pull/114446 Approved by: https://github.com/chenyang78, https://github.com/eellison |
||
|---|---|---|
| .. | ||
| aot_inductor | ||
| api | ||
| c10d | ||
| common | ||
| dist_autograd | ||
| jit | ||
| lazy | ||
| lite_interpreter_runtime | ||
| monitor | ||
| profiler | ||
| rpc | ||
| tensorexpr | ||
| __init__.py | ||