[modular] improve auto offload #14304
yiyixuxu
wants to merge
18 commits into
main
from
auto-offload-v2
[modular] auto offload: tunable memory reserve, OOM retry, non-disrup…
780e112c
[modular] drop stale "dynamic mode" wording from the offload test com…
d4b4edd1
Record what the auto-offloader does
34036c1f
Group the offload record table by decision
24460ae6
Split display helpers into components_manager_utils; drop activation …
b755e08d
Trim the offload record to consumed fields; hook reads available memo…
d2eb134a
Make offload record events pure moves
09d5ec7c
Drop OffloadRecord.summary(): the table is the record
d3c1e9d7
Address review: discover entry points via an apply_forward_hook marker
b3604203
Inline the OOM-retry helper into wrap_forward, its only caller
b313f79e
Shrink ComponentsManager to its used surface
0e512ddf
link issue to TODO
77615c07
Address review: record at call sites, one device rule, fewer moving p…
ba14c2a4
Overhaul the ComponentsManager test suite around real behavior
3f561491
Add slow modular-pipeline integration tests with declared offload beh…
29bd6e06
Restore ModularPipelineOffloadTesterMixin to its original form
e20dc87f
add set_offload_strategy, Peak column in offload record, and offloadi…
0750134d
yiyixuxu
commented
on 2026-07-31
yiyixuxu
commented
on 2026-07-31
make style
9a1c508e
Assignees
No one assigned
Labels
documentation
tests
size/L
modular-pipelines
utils
Milestone
Login to write a write a comment.
Login via GitHub