context_tracking: Restore preempted context state after preempt_schedule_irq()