#include <linux/kernel.h>
 #include <linux/mlx5/driver.h>
 #include <linux/mlx5/cmd.h>
+#include "mlx5_ib.h"
 #include "srq.h"
 
 static int get_pas_size(struct mlx5_srq_attr *in)
        in->db_record     = MLX5_GET64(srqc, srqc, dbr_addr);
 }
 
-struct mlx5_core_srq *mlx5_core_get_srq(struct mlx5_core_dev *dev, u32 srqn)
+struct mlx5_core_srq *mlx5_cmd_get_srq(struct mlx5_ib_dev *dev, u32 srqn)
 {
-       struct mlx5_srq_table *table = &dev->priv.srq_table;
+       struct mlx5_core_dev *mdev = dev->mdev;
+       struct mlx5_srq_table *table;
        struct mlx5_core_srq *srq;
 
+       table = &mdev->priv.srq_table;
+
        spin_lock(&table->lock);
 
        srq = radix_tree_lookup(&table->tree, srqn);
        return srq;
 }
 
-static int create_srq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int create_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                          struct mlx5_srq_attr *in)
 {
        u32 create_out[MLX5_ST_SZ_DW(create_srq_out)] = {0};
        MLX5_SET(create_srq_in, create_in, opcode,
                 MLX5_CMD_OP_CREATE_SRQ);
 
-       err = mlx5_cmd_exec(dev, create_in, inlen, create_out,
+       err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,
                            sizeof(create_out));
        kvfree(create_in);
        if (!err) {
        return err;
 }
 
-static int destroy_srq_cmd(struct mlx5_core_dev *dev,
-                          struct mlx5_core_srq *srq)
+static int destroy_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)
 {
        u32 srq_in[MLX5_ST_SZ_DW(destroy_srq_in)] = {0};
        u32 srq_out[MLX5_ST_SZ_DW(destroy_srq_out)] = {0};
        MLX5_SET(destroy_srq_in, srq_in, srqn, srq->srqn);
        MLX5_SET(destroy_srq_in, srq_in, uid, srq->uid);
 
-       return mlx5_cmd_exec(dev, srq_in, sizeof(srq_in),
-                            srq_out, sizeof(srq_out));
+       return mlx5_cmd_exec(dev->mdev, srq_in, sizeof(srq_in), srq_out,
+                            sizeof(srq_out));
 }
 
-static int arm_srq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int arm_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                       u16 lwm, int is_srq)
 {
        u32 srq_in[MLX5_ST_SZ_DW(arm_rq_in)] = {0};
        MLX5_SET(arm_rq_in, srq_in, lwm,      lwm);
        MLX5_SET(arm_rq_in, srq_in, uid, srq->uid);
 
-       return  mlx5_cmd_exec(dev, srq_in, sizeof(srq_in),
-                             srq_out, sizeof(srq_out));
+       return mlx5_cmd_exec(dev->mdev, srq_in, sizeof(srq_in), srq_out,
+                            sizeof(srq_out));
 }
 
-static int query_srq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int query_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                         struct mlx5_srq_attr *out)
 {
        u32 srq_in[MLX5_ST_SZ_DW(query_srq_in)] = {0};
        MLX5_SET(query_srq_in, srq_in, opcode,
                 MLX5_CMD_OP_QUERY_SRQ);
        MLX5_SET(query_srq_in, srq_in, srqn, srq->srqn);
-       err =  mlx5_cmd_exec(dev, srq_in, sizeof(srq_in),
-                            srq_out, MLX5_ST_SZ_BYTES(query_srq_out));
+       err = mlx5_cmd_exec(dev->mdev, srq_in, sizeof(srq_in), srq_out,
+                           MLX5_ST_SZ_BYTES(query_srq_out));
        if (err)
                goto out;
 
        return err;
 }
 
-static int create_xrc_srq_cmd(struct mlx5_core_dev *dev,
+static int create_xrc_srq_cmd(struct mlx5_ib_dev *dev,
                              struct mlx5_core_srq *srq,
                              struct mlx5_srq_attr *in)
 {
                 MLX5_CMD_OP_CREATE_XRC_SRQ);
 
        memset(create_out, 0, sizeof(create_out));
-       err = mlx5_cmd_exec(dev, create_in, inlen, create_out,
+       err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,
                            sizeof(create_out));
        if (err)
                goto out;
        return err;
 }
 
-static int destroy_xrc_srq_cmd(struct mlx5_core_dev *dev,
+static int destroy_xrc_srq_cmd(struct mlx5_ib_dev *dev,
                               struct mlx5_core_srq *srq)
 {
        u32 xrcsrq_in[MLX5_ST_SZ_DW(destroy_xrc_srq_in)]   = {0};
        MLX5_SET(destroy_xrc_srq_in, xrcsrq_in, xrc_srqn, srq->srqn);
        MLX5_SET(destroy_xrc_srq_in, xrcsrq_in, uid, srq->uid);
 
-       return mlx5_cmd_exec(dev, xrcsrq_in, sizeof(xrcsrq_in),
+       return mlx5_cmd_exec(dev->mdev, xrcsrq_in, sizeof(xrcsrq_in),
                             xrcsrq_out, sizeof(xrcsrq_out));
 }
 
-static int arm_xrc_srq_cmd(struct mlx5_core_dev *dev,
-                          struct mlx5_core_srq *srq, u16 lwm)
+static int arm_xrc_srq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
+                          u16 lwm)
 {
        u32 xrcsrq_in[MLX5_ST_SZ_DW(arm_xrc_srq_in)]   = {0};
        u32 xrcsrq_out[MLX5_ST_SZ_DW(arm_xrc_srq_out)] = {0};
        MLX5_SET(arm_xrc_srq_in, xrcsrq_in, lwm,      lwm);
        MLX5_SET(arm_xrc_srq_in, xrcsrq_in, uid, srq->uid);
 
-       return  mlx5_cmd_exec(dev, xrcsrq_in, sizeof(xrcsrq_in),
+       return  mlx5_cmd_exec(dev->mdev, xrcsrq_in, sizeof(xrcsrq_in),
                              xrcsrq_out, sizeof(xrcsrq_out));
 }
 
-static int query_xrc_srq_cmd(struct mlx5_core_dev *dev,
+static int query_xrc_srq_cmd(struct mlx5_ib_dev *dev,
                             struct mlx5_core_srq *srq,
                             struct mlx5_srq_attr *out)
 {
                 MLX5_CMD_OP_QUERY_XRC_SRQ);
        MLX5_SET(query_xrc_srq_in, xrcsrq_in, xrc_srqn, srq->srqn);
 
-       err =  mlx5_cmd_exec(dev, xrcsrq_in, sizeof(xrcsrq_in), xrcsrq_out,
-                            MLX5_ST_SZ_BYTES(query_xrc_srq_out));
+       err = mlx5_cmd_exec(dev->mdev, xrcsrq_in, sizeof(xrcsrq_in),
+                           xrcsrq_out, MLX5_ST_SZ_BYTES(query_xrc_srq_out));
        if (err)
                goto out;
 
        return err;
 }
 
-static int mlx5_core_create_rmp(struct mlx5_core_dev *dev, u32 *in, int inlen,
+static int mlx5_core_create_rmp(struct mlx5_ib_dev *dev, u32 *in, int inlen,
                                u32 *rmpn)
 {
        u32 out[MLX5_ST_SZ_DW(create_rmp_out)] = { 0 };
        int err;
 
        MLX5_SET(create_rmp_in, in, opcode, MLX5_CMD_OP_CREATE_RMP);
-       err = mlx5_cmd_exec(dev, in, inlen, out, sizeof(out));
+       err = mlx5_cmd_exec(dev->mdev, in, inlen, out, sizeof(out));
        if (!err)
                *rmpn = MLX5_GET(create_rmp_out, out, rmpn);
 
        return err;
 }
 
-static int mlx5_core_modify_rmp(struct mlx5_core_dev *dev, u32 *in, int inlen)
+static int mlx5_core_modify_rmp(struct mlx5_ib_dev *dev, u32 *in, int inlen)
 {
        u32 out[MLX5_ST_SZ_DW(modify_rmp_out)] = {0};
 
        MLX5_SET(modify_rmp_in, in, opcode, MLX5_CMD_OP_MODIFY_RMP);
-       return mlx5_cmd_exec(dev, in, inlen, out, sizeof(out));
+       return mlx5_cmd_exec(dev->mdev, in, inlen, out, sizeof(out));
 }
 
-static int mlx5_core_query_rmp(struct mlx5_core_dev *dev, u32 rmpn, u32 *out)
+static int mlx5_core_query_rmp(struct mlx5_ib_dev *dev, u32 rmpn, u32 *out)
 {
        u32 in[MLX5_ST_SZ_DW(query_rmp_in)] = {0};
        int outlen = MLX5_ST_SZ_BYTES(query_rmp_out);
 
        MLX5_SET(query_rmp_in, in, opcode, MLX5_CMD_OP_QUERY_RMP);
        MLX5_SET(query_rmp_in, in, rmpn,   rmpn);
-       return mlx5_cmd_exec(dev, in, sizeof(in), out, outlen);
+       return mlx5_cmd_exec(dev->mdev, in, sizeof(in), out, outlen);
 }
 
-static int create_rmp_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int create_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                          struct mlx5_srq_attr *in)
 {
        void *create_in;
        return err;
 }
 
-static int destroy_rmp_cmd(struct mlx5_core_dev *dev,
-                          struct mlx5_core_srq *srq)
+static int destroy_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)
 {
        u32 in[MLX5_ST_SZ_DW(destroy_rmp_in)]   = {};
        u32 out[MLX5_ST_SZ_DW(destroy_rmp_out)] = {};
        MLX5_SET(destroy_rmp_in, in, opcode, MLX5_CMD_OP_DESTROY_RMP);
        MLX5_SET(destroy_rmp_in, in, rmpn, srq->srqn);
        MLX5_SET(destroy_rmp_in, in, uid, srq->uid);
-       return mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
+       return mlx5_cmd_exec(dev->mdev, in, sizeof(in), out, sizeof(out));
 }
 
-static int arm_rmp_cmd(struct mlx5_core_dev *dev,
-                      struct mlx5_core_srq *srq,
+static int arm_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                       u16 lwm)
 {
        void *in;
        return err;
 }
 
-static int query_rmp_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int query_rmp_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                         struct mlx5_srq_attr *out)
 {
        u32 *rmp_out;
        return err;
 }
 
-static int create_xrq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int create_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                          struct mlx5_srq_attr *in)
 {
        u32 create_out[MLX5_ST_SZ_DW(create_xrq_out)] = {0};
        MLX5_SET(xrqc, xrqc, cqn, in->cqn);
        MLX5_SET(create_xrq_in, create_in, opcode, MLX5_CMD_OP_CREATE_XRQ);
        MLX5_SET(create_xrq_in, create_in, uid, in->uid);
-       err = mlx5_cmd_exec(dev, create_in, inlen, create_out,
+       err = mlx5_cmd_exec(dev->mdev, create_in, inlen, create_out,
                            sizeof(create_out));
        kvfree(create_in);
        if (!err) {
        return err;
 }
 
-static int destroy_xrq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq)
+static int destroy_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)
 {
        u32 in[MLX5_ST_SZ_DW(destroy_xrq_in)] = {0};
        u32 out[MLX5_ST_SZ_DW(destroy_xrq_out)] = {0};
        MLX5_SET(destroy_xrq_in, in, xrqn,   srq->srqn);
        MLX5_SET(destroy_xrq_in, in, uid, srq->uid);
 
-       return mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
+       return mlx5_cmd_exec(dev->mdev, in, sizeof(in), out, sizeof(out));
 }
 
-static int arm_xrq_cmd(struct mlx5_core_dev *dev,
+static int arm_xrq_cmd(struct mlx5_ib_dev *dev,
                       struct mlx5_core_srq *srq,
                       u16 lwm)
 {
        MLX5_SET(arm_rq_in, in, lwm,        lwm);
        MLX5_SET(arm_rq_in, in, uid, srq->uid);
 
-       return mlx5_cmd_exec(dev, in, sizeof(in), out, sizeof(out));
+       return mlx5_cmd_exec(dev->mdev, in, sizeof(in), out, sizeof(out));
 }
 
-static int query_xrq_cmd(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
+static int query_xrq_cmd(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                         struct mlx5_srq_attr *out)
 {
        u32 in[MLX5_ST_SZ_DW(query_xrq_in)] = {0};
        MLX5_SET(query_xrq_in, in, opcode, MLX5_CMD_OP_QUERY_XRQ);
        MLX5_SET(query_xrq_in, in, xrqn, srq->srqn);
 
-       err = mlx5_cmd_exec(dev, in, sizeof(in), xrq_out, outlen);
+       err = mlx5_cmd_exec(dev->mdev, in, sizeof(in), xrq_out, outlen);
        if (err)
                goto out;
 
        return err;
 }
 
-static int create_srq_split(struct mlx5_core_dev *dev,
-                           struct mlx5_core_srq *srq,
+static int create_srq_split(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
                            struct mlx5_srq_attr *in)
 {
-       if (!dev->issi)
+       if (!dev->mdev->issi)
                return create_srq_cmd(dev, srq, in);
        switch (srq->common.res) {
        case MLX5_RES_XSRQ:
        }
 }
 
-static int destroy_srq_split(struct mlx5_core_dev *dev,
-                            struct mlx5_core_srq *srq)
+static int destroy_srq_split(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)
 {
-       if (!dev->issi)
+       if (!dev->mdev->issi)
                return destroy_srq_cmd(dev, srq);
        switch (srq->common.res) {
        case MLX5_RES_XSRQ:
        }
 }
 
-int mlx5_core_create_srq(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
-                        struct mlx5_srq_attr *in)
+int mlx5_cmd_create_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
+                       struct mlx5_srq_attr *in)
 {
+       struct mlx5_core_dev *mdev = dev->mdev;
+       struct mlx5_srq_table *table;
        int err;
-       struct mlx5_srq_table *table = &dev->priv.srq_table;
+
+       table = &mdev->priv.srq_table;
 
        switch (in->type) {
        case IB_SRQT_XRC:
        return err;
 }
 
-int mlx5_core_destroy_srq(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq)
+int mlx5_cmd_destroy_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq)
 {
-       struct mlx5_srq_table *table = &dev->priv.srq_table;
+       struct mlx5_core_dev *mdev = dev->mdev;
+       struct mlx5_srq_table *table;
        struct mlx5_core_srq *tmp;
        int err;
 
+       table = &mdev->priv.srq_table;
+
        spin_lock_irq(&table->lock);
        tmp = radix_tree_delete(&table->tree, srq->srqn);
        spin_unlock_irq(&table->lock);
        return 0;
 }
 
-int mlx5_core_query_srq(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
-                       struct mlx5_srq_attr *out)
+int mlx5_cmd_query_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
+                      struct mlx5_srq_attr *out)
 {
-       if (!dev->issi)
+       if (!dev->mdev->issi)
                return query_srq_cmd(dev, srq, out);
        switch (srq->common.res) {
        case MLX5_RES_XSRQ:
        }
 }
 
-int mlx5_core_arm_srq(struct mlx5_core_dev *dev, struct mlx5_core_srq *srq,
-                     u16 lwm, int is_srq)
+int mlx5_cmd_arm_srq(struct mlx5_ib_dev *dev, struct mlx5_core_srq *srq,
+                    u16 lwm, int is_srq)
 {
-       if (!dev->issi)
+       if (!dev->mdev->issi)
                return arm_srq_cmd(dev, srq, lwm, is_srq);
        switch (srq->common.res) {
        case MLX5_RES_XSRQ: