2 files changed, 6 insertions, 1 deletions

diff --git a/arch/powerpc/platforms/cell/spufs/context.c b/arch/powerpc/platforms/cell/spufs/context.c
index 6694f86d7000..9cb081c26e71 100644
--- a/arch/powerpc/platforms/cell/spufs/context.c
+++ b/arch/powerpc/platforms/cell/spufs/context.c
@@ -59,7 +59,8 @@ struct spu_context *alloc_spu_context(struct spu_gang *gang)
         INIT_LIST_HEAD(&ctx->aff_list);
         if (gang)
                 spu_gang_add_ctx(gang, ctx);
-        ctx->cpus_allowed = current->cpus_allowed;
+
+        __spu_update_sched_info(ctx);
         spu_set_timeslice(ctx);
         ctx->stats.util_state = SPU_UTIL_IDLE_LOADED;
 
diff --git a/arch/powerpc/platforms/cell/spufs/run.c b/arch/powerpc/platforms/cell/spufs/run.c
index 0b50fa5cb39d..6abdd8fe2098 100644
--- a/arch/powerpc/platforms/cell/spufs/run.c
+++ b/arch/powerpc/platforms/cell/spufs/run.c
@@ -312,6 +312,7 @@ long spufs_run_spu(struct spu_context *ctx, u32 *npc, u32 *event)
         spu_acquire(ctx);
         if (ctx->state == SPU_STATE_SAVED) {
                 __spu_update_sched_info(ctx);
+                spu_set_timeslice(ctx);
 
                 ret = spu_activate(ctx, 0);
                 if (ret) {
@@ -322,6 +323,9 @@ long spufs_run_spu(struct spu_context *ctx, u32 *npc, u32 *event)
                 /*
                  * We have to update the scheduling priority under active_mutex
                  * to protect against find_victim().
+                 *
+                 * No need to update the timeslice ASAP, it will get updated
+                 * once the current one has expired.
                  */
                 spu_update_sched_info(ctx);
         }