kfree(e);
 }
 
+/*
+ * Use the default elevator specified by config boot param for non-mq devices,
+ * or by config option.  Don't try to load modules as we could be running off
+ * async and request_module() isn't allowed from async.
+ */
 int elevator_init(struct request_queue *q)
 {
        struct elevator_type *e = NULL;
        if (unlikely(q->elevator))
                goto out_unlock;
 
-       /*
-        * Use the default elevator specified by config boot param for
-        * non-mq devices, or by config option. Don't try to load modules
-        * as we could be running off async and request_module() isn't
-        * allowed from async.
-        */
-       if (!q->mq_ops && *chosen_elevator) {
+       if (*chosen_elevator) {
                e = elevator_get(q, chosen_elevator, false);
                if (!e)
                        printk(KERN_ERR "I/O scheduler %s not found\n",
                                                        chosen_elevator);
        }
 
+       if (!e)
+               e = elevator_get(q, CONFIG_DEFAULT_IOSCHED, false);
        if (!e) {
-               /*
-                * For blk-mq devices, we default to using mq-deadline,
-                * if available, for single queue devices. If deadline
-                * isn't available OR we have multiple queues, default
-                * to "none".
-                */
-               if (q->mq_ops) {
-                       if (q->nr_hw_queues == 1)
-                               e = elevator_get(q, "mq-deadline", false);
-                       if (!e)
-                               goto out_unlock;
-               } else
-                       e = elevator_get(q, CONFIG_DEFAULT_IOSCHED, false);
-
-               if (!e) {
-                       printk(KERN_ERR
-                               "Default I/O scheduler not found. " \
-                               "Using noop.\n");
-                       e = elevator_get(q, "noop", false);
-               }
+               printk(KERN_ERR
+                       "Default I/O scheduler not found. Using noop.\n");
+               e = elevator_get(q, "noop", false);
        }
 
-       if (e->uses_mq)
-               err = blk_mq_init_sched(q, e);
-       else
-               err = e->ops.sq.elevator_init_fn(q, e);
+       err = e->ops.sq.elevator_init_fn(q, e);
        if (err)
                elevator_put(e);
 out_unlock:
        return ret;
 }
 
+/*
+ * For blk-mq devices, we default to using mq-deadline, if available, for single
+ * queue devices.  If deadline isn't available OR we have multiple queues,
+ * default to "none".
+ */
+int elevator_init_mq(struct request_queue *q)
+{
+       struct elevator_type *e;
+       int err = 0;
+
+       if (q->nr_hw_queues != 1)
+               return 0;
+
+       /*
+        * q->sysfs_lock must be held to provide mutual exclusion between
+        * elevator_switch() and here.
+        */
+       mutex_lock(&q->sysfs_lock);
+       if (unlikely(q->elevator))
+               goto out_unlock;
+
+       e = elevator_get(q, "mq-deadline", false);
+       if (!e)
+               goto out_unlock;
+
+       err = blk_mq_init_sched(q, e);
+       if (err)
+               elevator_put(e);
+out_unlock:
+       mutex_unlock(&q->sysfs_lock);
+       return err;
+}
+
+
 /*
  * switch to new_e io scheduler. be careful not to introduce deadlocks -
  * we don't free the old io scheduler, before we have allocated what we