time_bench: proper shutdown of kthreads

Signed-off-by: Jesper Dangaard Brouer <[email protected]>
apalos · Jan 20, 2020 · ed04e93 · ed04e93
1 parent 4e29a34
commit ed04e93
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/kernel/lib/time_bench.c b/kernel/lib/time_bench.c
@@ -295,8 +295,12 @@ static int invoke_test_on_cpu_func(void *private)
 
 	/* End test */
 	atomic_dec(&sync->nr_tests_running);
-	set_current_state(TASK_UNINTERRUPTIBLE);
-	schedule();
+	/*  Wait for kthread_stop() telling us to stop */
+	while (!kthread_should_stop()) {
+		set_current_state(TASK_INTERRUPTIBLE);
+		schedule();
+	}
+	__set_current_state(TASK_RUNNING);
 	return 0;
 }