[POWERPC] spu sched: ensure preempted threads are put back on the runqueue, part2

To not lose a spu thread we need to make sure it always gets put back
on the runqueue.  In find_victim aswell as in the scheduler tick as done
in the previous patch.

Signed-off-by: Christoph Hellwig <hch@lst.de>
Signed-off-by: Arnd Bergmann <arnd.bergmann@de.ibm.com>

1 files changed, 6 insertions, 0 deletions

diff --git a/arch/powerpc/platforms/cell/spufs/sched.c b/arch/powerpc/platforms/cell/spufs/sched.c
index 5149dff65c9e..405a0555d75c 100644
--- a/arch/powerpc/platforms/cell/spufs/sched.c
+++ b/arch/powerpc/platforms/cell/spufs/sched.c
@@ -391,6 +391,12 @@ static struct spu *find_victim(struct spu_context *ctx)
                         }
                         spu_unbind_context(spu, victim);
                         mutex_unlock(&victim->state_mutex);
+                        /*
+                         * We need to break out of the wait loop in spu_run
+                         * manually to ensure this context gets put on the
+                         * runqueue again ASAP.
+                         */
+                        wake_up(&victim->stop_wq);
                         return spu;
                 }
         }