IB/mlx5: Refactor UMR to have its own context struct