mm: writeback tweaks during screen on/off

Signed-off-by: zwliew <zhaoweiliew@gmail.com>