extern void _drbd_thread_stop(struct drbd_thread *thi, int restart, int wait);
 extern char *drbd_task_to_thread_name(struct drbd_conf *mdev, struct task_struct *task);
 #ifdef CONFIG_SMP
-extern void drbd_thread_current_set_cpu(struct drbd_conf *mdev);
+extern void drbd_thread_current_set_cpu(struct drbd_conf *mdev, struct drbd_thread *thi);
 extern void drbd_calc_cpu_mask(struct drbd_conf *mdev);
 #else
-#define drbd_thread_current_set_cpu(A) ({})
+#define drbd_thread_current_set_cpu(A, B) ({})
 #define drbd_calc_cpu_mask(A) ({})
 #endif
 extern void drbd_free_resources(struct drbd_conf *mdev);
 
 /**
  * drbd_thread_current_set_cpu() - modifies the cpu mask of the _current_ thread
  * @mdev:      DRBD device.
+ * @thi:       drbd_thread object
  *
  * call in the "main loop" of _all_ threads, no need for any mutex, current won't die
  * prematurely.
  */
-void drbd_thread_current_set_cpu(struct drbd_conf *mdev)
+void drbd_thread_current_set_cpu(struct drbd_conf *mdev, struct drbd_thread *thi)
 {
        struct task_struct *p = current;
-       struct drbd_thread *thi = drbd_task_to_thread(mdev, p);
 
-       if (!expect(thi != NULL))
-               return;
        if (!thi->reset_cpu_mask)
                return;
        thi->reset_cpu_mask = 0;
 
        int rv;
 
        while (get_t_state(&mdev->tconn->receiver) == RUNNING) {
-               drbd_thread_current_set_cpu(mdev);
+               drbd_thread_current_set_cpu(mdev, &mdev->tconn->receiver);
                if (!drbd_recv_header(mdev, &cmd, &packet_size))
                        goto err_out;
 
        current->rt_priority = 2;    /* more important than all other tasks */
 
        while (get_t_state(thi) == RUNNING) {
-               drbd_thread_current_set_cpu(mdev);
+               drbd_thread_current_set_cpu(mdev, thi);
                if (test_and_clear_bit(SEND_PING, &mdev->flags)) {
                        if (!drbd_send_ping(mdev)) {
                                dev_err(DEV, "drbd_send_ping has failed\n");
 
        sprintf(current->comm, "drbd%d_worker", mdev_to_minor(mdev));
 
        while (get_t_state(thi) == RUNNING) {
-               drbd_thread_current_set_cpu(mdev);
+               drbd_thread_current_set_cpu(mdev, thi);
 
                if (down_trylock(&mdev->tconn->data.work.s)) {
                        mutex_lock(&mdev->tconn->data.mutex);