1 files changed, 10 insertions, 6 deletions

diff --git a/kernel/rcutorture.c b/kernel/rcutorture.c
index fd599829e72a..33acc424667e 100644
--- a/kernel/rcutorture.c
+++ b/kernel/rcutorture.c
@@ -45,6 +45,7 @@
 #include <linux/byteorder/swabb.h>
 #include <linux/stat.h>
 #include <linux/srcu.h>
+#include <linux/slab.h>
 
 MODULE_LICENSE("GPL");
 MODULE_AUTHOR("Paul E. McKenney <paulmck@us.ibm.com> and "
@@ -723,9 +724,10 @@ static int rcu_idle_cpu;	/* Force all torture tasks off this CPU */
  */
 static void rcu_torture_shuffle_tasks(void)
 {
-        cpumask_t tmp_mask = CPU_MASK_ALL;
+        cpumask_t tmp_mask;
         int i;
 
+        cpus_setall(tmp_mask);
         get_online_cpus();
 
         /* No point in shuffling if there is only one online CPU (ex: UP) */
@@ -737,25 +739,27 @@ static void rcu_torture_shuffle_tasks(void)
         if (rcu_idle_cpu != -1)
                 cpu_clear(rcu_idle_cpu, tmp_mask);
 
-        set_cpus_allowed(current, tmp_mask);
+        set_cpus_allowed_ptr(current, &tmp_mask);
 
         if (reader_tasks) {
                 for (i = 0; i < nrealreaders; i++)
                         if (reader_tasks[i])
-                                set_cpus_allowed(reader_tasks[i], tmp_mask);
+                                set_cpus_allowed_ptr(reader_tasks[i],
+                                                     &tmp_mask);
         }
 
         if (fakewriter_tasks) {
                 for (i = 0; i < nfakewriters; i++)
                         if (fakewriter_tasks[i])
-                                set_cpus_allowed(fakewriter_tasks[i], tmp_mask);
+                                set_cpus_allowed_ptr(fakewriter_tasks[i],
+                                                     &tmp_mask);
         }
 
         if (writer_task)
-                set_cpus_allowed(writer_task, tmp_mask);
+                set_cpus_allowed_ptr(writer_task, &tmp_mask);
 
         if (stats_task)
-                set_cpus_allowed(stats_task, tmp_mask);
+                set_cpus_allowed_ptr(stats_task, &tmp_mask);
 
         if (rcu_idle_cpu == -1)
                 rcu_idle_cpu = num_online_cpus() - 1;