put_cpu only after queueing works.

diff --git a/block_wq.c b/block_wq.c
index ae6d8ab..1b3fc0b 100644 (file)
--- a/block_wq.c
+++ b/block_wq.c
@@ -53,9 +53,9 @@ static int block_wq_init(void)
        if (!block_wq)
                return -ENOMEM;
        printk(KERN_INFO "Queueing task in CPU %d\n", get_cpu());
-       put_cpu();
        queue_work(block_wq, &block_work);
        queue_work(block_wq, &print_work);
+       put_cpu();
        return 0;
 }