1 files changed, 6 insertions, 5 deletions

diff --git a/drivers/net/ethernet/mellanox/mlx5/core/en_rep.c b/drivers/net/ethernet/mellanox/mlx5/core/en_rep.c
index 363d8dcb7f17..0273c233bc85 100644
--- a/drivers/net/ethernet/mellanox/mlx5/core/en_rep.c
+++ b/drivers/net/ethernet/mellanox/mlx5/core/en_rep.c
@@ -668,7 +668,6 @@ static int mlx5e_rep_open(struct net_device *dev)
         struct mlx5e_priv *priv = netdev_priv(dev);
         struct mlx5e_rep_priv *rpriv = priv->ppriv;
         struct mlx5_eswitch_rep *rep = rpriv->rep;
-        struct mlx5_eswitch *esw = priv->mdev->priv.eswitch;
         int err;
 
         mutex_lock(&priv->state_lock);
@@ -676,8 +675,9 @@ static int mlx5e_rep_open(struct net_device *dev)
         if (err)
                 goto unlock;
 
-        if (!mlx5_eswitch_set_vport_state(esw, rep->vport,
-                                          MLX5_ESW_VPORT_ADMIN_STATE_UP))
+        if (!mlx5_modify_vport_admin_state(priv->mdev,
+                        MLX5_QUERY_VPORT_STATE_IN_OP_MOD_ESW_VPORT,
+                        rep->vport, MLX5_ESW_VPORT_ADMIN_STATE_UP))
                 netif_carrier_on(dev);
 
 unlock:
@@ -690,11 +690,12 @@ static int mlx5e_rep_close(struct net_device *dev)
         struct mlx5e_priv *priv = netdev_priv(dev);
         struct mlx5e_rep_priv *rpriv = priv->ppriv;
         struct mlx5_eswitch_rep *rep = rpriv->rep;
-        struct mlx5_eswitch *esw = priv->mdev->priv.eswitch;
         int ret;
 
         mutex_lock(&priv->state_lock);
-        (void)mlx5_eswitch_set_vport_state(esw, rep->vport, MLX5_ESW_VPORT_ADMIN_STATE_DOWN);
+        mlx5_modify_vport_admin_state(priv->mdev,
+                        MLX5_QUERY_VPORT_STATE_IN_OP_MOD_ESW_VPORT,
+                        rep->vport, MLX5_ESW_VPORT_ADMIN_STATE_DOWN);
         ret = mlx5e_close_locked(dev);
         mutex_unlock(&priv->state_lock);
         return ret;