sched: Allow wait_on_bit_action() functions to support a timeout