X-Git-Url: http://git.monstr.eu/?a=blobdiff_plain;f=kernel%2Ftorture.c;h=bb8f411c974b8d51297708f51dedc41c46c11757;hb=0d290223a6c77107b1c3988959e49279a8dafaba;hp=0a315c387bedb25a630a61577d4e7b01471d1466;hpb=33197bd3e82f5c60487e53d4a291dc2e6031833f;p=linux-2.6-microblaze.git

diff --git a/kernel/torture.c b/kernel/torture.c
index 0a315c387bed..bb8f411c974b 100644
--- a/kernel/torture.c
+++ b/kernel/torture.c
@@ -521,11 +521,11 @@ static void torture_shuffle_tasks(void)
 	struct shuffle_task *stp;
 
 	cpumask_setall(shuffle_tmp_mask);
-	get_online_cpus();
+	cpus_read_lock();
 
 	/* No point in shuffling if there is only one online CPU (ex: UP) */
 	if (num_online_cpus() == 1) {
-		put_online_cpus();
+		cpus_read_unlock();
 		return;
 	}
 
@@ -541,7 +541,7 @@ static void torture_shuffle_tasks(void)
 		set_cpus_allowed_ptr(stp->st_t, shuffle_tmp_mask);
 	mutex_unlock(&shuffle_task_mutex);
 
-	put_online_cpus();
+	cpus_read_unlock();
 }
 
 /* Shuffle tasks across CPUs, with the intent of allowing each CPU in the