os: sched_enter: shorten critical section

We can release the runq_lock as soon as we've marked our process as
 TASK_RUNNING, since any other CPUs will ignore a RUNNING runq entry.