From: Maor Gottlieb <ma...@nvidia.com>

Add two functions to allocate and deallocate MEMIC operations
by using the MODIFY_MEMIC command.

Signed-off-by: Maor Gottlieb <ma...@nvidia.com>
Signed-off-by: Leon Romanovsky <leo...@nvidia.com>
---
 drivers/infiniband/hw/mlx5/dm.c | 38 +++++++++++++++++++++++++++++++++
 drivers/infiniband/hw/mlx5/dm.h |  2 ++
 2 files changed, 40 insertions(+)

diff --git a/drivers/infiniband/hw/mlx5/dm.c b/drivers/infiniband/hw/mlx5/dm.c
index 3d39d93625ad..97a925d43312 100644
--- a/drivers/infiniband/hw/mlx5/dm.c
+++ b/drivers/infiniband/hw/mlx5/dm.c
@@ -111,6 +111,44 @@ void mlx5_cmd_dealloc_memic(struct mlx5_dm *dm, 
phys_addr_t addr,
        spin_unlock(&dm->lock);
 }

+void mlx5_cmd_dealloc_memic_op(struct mlx5_dm *dm, phys_addr_t addr,
+                              u8 operation)
+{
+       u32 in[MLX5_ST_SZ_DW(modify_memic_in)] = {};
+       struct mlx5_core_dev *dev = dm->dev;
+
+       MLX5_SET(modify_memic_in, in, opcode, MLX5_CMD_OP_MODIFY_MEMIC);
+       MLX5_SET(modify_memic_in, in, op_mod, MLX5_MODIFY_MEMIC_OP_MOD_DEALLOC);
+       MLX5_SET(modify_memic_in, in, memic_operation_type,
+                MLX5_MODIFY_MEMIC_OP_MOD_ALLOC);
+       MLX5_SET64(modify_memic_in, in, memic_start_addr, addr - dev->bar_addr);
+
+       mlx5_cmd_exec_in(dev, modify_memic, in);
+}
+
+static int mlx5_cmd_alloc_memic_op(struct mlx5_dm *dm, phys_addr_t addr,
+                                  u8 operation, phys_addr_t *op_addr)
+{
+       u32 out[MLX5_ST_SZ_DW(modify_memic_out)] = {};
+       u32 in[MLX5_ST_SZ_DW(modify_memic_in)] = {};
+       struct mlx5_core_dev *dev = dm->dev;
+       int err;
+
+       MLX5_SET(modify_memic_in, in, opcode, MLX5_CMD_OP_MODIFY_MEMIC);
+       MLX5_SET(modify_memic_in, in, op_mod, MLX5_MODIFY_MEMIC_OP_MOD_ALLOC);
+       MLX5_SET(modify_memic_in, in, memic_operation_type,
+                MLX5_MODIFY_MEMIC_OP_MOD_ALLOC);
+       MLX5_SET64(modify_memic_in, in, memic_start_addr, addr - dev->bar_addr);
+
+       err = mlx5_cmd_exec_inout(dev, modify_memic, in, out);
+       if (err)
+               return err;
+
+       *op_addr = dev->bar_addr +
+                  MLX5_GET64(modify_memic_out, out, memic_operation_addr);
+       return 0;
+}
+
 static int add_dm_mmap_entry(struct ib_ucontext *context,
                             struct mlx5_ib_dm *mdm, u64 address)
 {
diff --git a/drivers/infiniband/hw/mlx5/dm.h b/drivers/infiniband/hw/mlx5/dm.h
index dbef67e38731..adb39d3d8131 100644
--- a/drivers/infiniband/hw/mlx5/dm.h
+++ b/drivers/infiniband/hw/mlx5/dm.h
@@ -10,5 +10,7 @@

 void mlx5_cmd_dealloc_memic(struct mlx5_dm *dm, phys_addr_t addr,
                            u64 length);
+void mlx5_cmd_dealloc_memic_op(struct mlx5_dm *dm, phys_addr_t addr,
+                              u8 operation);

 #endif /* _MLX5_IB_DM_H */
--
2.30.2

Reply via email to