#include "en/devlink.h"
 
-int mlx5e_devlink_phy_port_register(struct net_device *dev)
+int mlx5e_devlink_port_register(struct net_device *netdev)
 {
+       struct mlx5_core_dev *dev;
        struct mlx5e_priv *priv;
        struct devlink *devlink;
        int err;
 
-       priv = netdev_priv(dev);
-       devlink = priv_to_devlink(priv->mdev);
-
-       devlink_port_attrs_set(&priv->dl_phy_port,
-                              DEVLINK_PORT_FLAVOUR_PHYSICAL,
-                              PCI_FUNC(priv->mdev->pdev->devfn),
-                              false, 0,
-                              NULL, 0);
-       err = devlink_port_register(devlink, &priv->dl_phy_port, 1);
+       priv = netdev_priv(netdev);
+       dev = priv->mdev;
+
+       if (mlx5_core_is_pf(dev))
+               devlink_port_attrs_set(&priv->dl_port,
+                                      DEVLINK_PORT_FLAVOUR_PHYSICAL,
+                                      PCI_FUNC(dev->pdev->devfn),
+                                      false, 0,
+                                      NULL, 0);
+       else
+               devlink_port_attrs_set(&priv->dl_port,
+                                      DEVLINK_PORT_FLAVOUR_VIRTUAL,
+                                      0, false, 0, NULL, 0);
+
+       devlink = priv_to_devlink(dev);
+       err = devlink_port_register(devlink, &priv->dl_port, 1);
        if (err)
                return err;
-       devlink_port_type_eth_set(&priv->dl_phy_port, dev);
+       devlink_port_type_eth_set(&priv->dl_port, netdev);
        return 0;
 }
 
-void mlx5e_devlink_phy_port_unregister(struct mlx5e_priv *priv)
+void mlx5e_devlink_port_unregister(struct mlx5e_priv *priv)
 {
-       devlink_port_unregister(&priv->dl_phy_port);
+       devlink_port_unregister(&priv->dl_port);
 }
 
-struct devlink_port *mlx5e_get_devlink_phy_port(struct net_device *dev)
+struct devlink_port *mlx5e_get_devlink_port(struct net_device *dev)
 {
        struct mlx5e_priv *priv = netdev_priv(dev);
 
-       return &priv->dl_phy_port;
+       return &priv->dl_port;
 }
-
 
 #include <net/devlink.h>
 #include "en.h"
 
-int mlx5e_devlink_phy_port_register(struct net_device *dev);
-void mlx5e_devlink_phy_port_unregister(struct mlx5e_priv *priv);
-struct devlink_port *mlx5e_get_devlink_phy_port(struct net_device *dev);
+int mlx5e_devlink_port_register(struct net_device *dev);
+void mlx5e_devlink_port_unregister(struct mlx5e_priv *priv);
+struct devlink_port *mlx5e_get_devlink_port(struct net_device *dev);
 
 #endif
-
 
        .ndo_set_vf_link_state   = mlx5e_set_vf_link_state,
        .ndo_get_vf_stats        = mlx5e_get_vf_stats,
 #endif
-       .ndo_get_devlink_port    = mlx5e_get_devlink_phy_port,
+       .ndo_get_devlink_port    = mlx5e_get_devlink_port,
 };
 
 static int mlx5e_check_required_hca_cap(struct mlx5_core_dev *mdev)
                goto err_detach;
        }
 
-       err = mlx5e_devlink_phy_port_register(netdev);
+       err = mlx5e_devlink_port_register(netdev);
        if (err) {
                mlx5_core_err(mdev, "mlx5e_devlink_phy_port_register failed, %d\n", err);
                goto err_unregister_netdev;
 #ifdef CONFIG_MLX5_CORE_EN_DCB
        mlx5e_dcbnl_delete_app(priv);
 #endif
-       mlx5e_devlink_phy_port_unregister(priv);
+       mlx5e_devlink_port_unregister(priv);
        unregister_netdev(priv->netdev);
        mlx5e_detach(mdev, vpriv);
        mlx5e_destroy_netdev(priv);