1 files changed, 11 insertions, 1 deletions

diff --git a/drivers/infiniband/hw/mlx5/main.c b/drivers/infiniband/hw/mlx5/main.c
index edd8b8741846..5acf346e048e 100644
--- a/drivers/infiniband/hw/mlx5/main.c
+++ b/drivers/infiniband/hw/mlx5/main.c
@@ -284,7 +284,7 @@ __be16 mlx5_get_roce_udp_sport(struct mlx5_ib_dev *dev, u8 port_num,
 
 static int mlx5_use_mad_ifc(struct mlx5_ib_dev *dev)
 {
-        return !dev->mdev->issi;
+        return !MLX5_CAP_GEN(dev->mdev, ib_virt);
 }
 
 enum {
@@ -563,6 +563,9 @@ static int mlx5_ib_query_device(struct ib_device *ibdev,
         if (MLX5_CAP_GEN(mdev, cd))
                 props->device_cap_flags |= IB_DEVICE_CROSS_CHANNEL;
 
+        if (!mlx5_core_is_pf(mdev))
+                props->device_cap_flags |= IB_DEVICE_VIRTUAL_FUNCTION;
+
         return 0;
 }
 
@@ -700,6 +703,7 @@ static int mlx5_query_hca_port(struct ib_device *ibdev, u8 port,
         props->qkey_viol_cntr   = rep->qkey_violation_counter;
         props->subnet_timeout   = rep->subnet_timeout;
         props->init_type_reply  = rep->init_type_reply;
+        props->grh_required     = rep->grh_required;
 
         err = mlx5_query_port_link_width_oper(mdev, &ib_link_width_oper, port);
         if (err)
@@ -2350,6 +2354,12 @@ static void *mlx5_ib_add(struct mlx5_core_dev *mdev)
         dev->ib_dev.map_mr_sg           = mlx5_ib_map_mr_sg;
         dev->ib_dev.check_mr_status     = mlx5_ib_check_mr_status;
         dev->ib_dev.get_port_immutable  = mlx5_port_immutable;
+        if (mlx5_core_is_pf(mdev)) {
+                dev->ib_dev.get_vf_config       = mlx5_ib_get_vf_config;
+                dev->ib_dev.set_vf_link_state   = mlx5_ib_set_vf_link_state;
+                dev->ib_dev.get_vf_stats        = mlx5_ib_get_vf_stats;
+                dev->ib_dev.set_vf_guid         = mlx5_ib_set_vf_guid;
+        }
 
         mlx5_ib_internal_fill_odp_caps(dev);