Skip to content

Make enable_ssd_writeback outer check in FeatureEvict#5924

Open
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D108950148
Open

Make enable_ssd_writeback outer check in FeatureEvict#5924
lizhe-ji wants to merge 1 commit into
pytorch:mainfrom
lizhe-ji:export-D108950148

Conversation

@lizhe-ji

Copy link
Copy Markdown

Summary: Resolves reviewer comment in D107491725 to add explicit flag control for SSD eviction writeback instead of relying on writeback_callback_ presence. Nest enable_ssd_writeback as upper level if condition in start_training_eviction_loop and start_inference_eviction_loop, with inner check for writeback_callback and dirty state or non-empty batch. This makes SSD-related eviction behavior controlled by explicit config flag rather than callback truthiness.

Differential Revision: D108950148

Summary: Resolves reviewer comment in D107491725 to add explicit flag control for SSD eviction writeback instead of relying on writeback_callback_ presence. Nest `enable_ssd_writeback` as upper level `if` condition in `start_training_eviction_loop` and `start_inference_eviction_loop`, with inner check for `writeback_callback` and dirty state or non-empty batch. This makes SSD-related eviction behavior controlled by explicit config flag rather than callback truthiness.

Differential Revision: D108950148
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant