Also move wake_up_locked() to be with the related functions
Signed-off-by: NMatthew Wilcox <willy@linux.intel.com>