vfio/mlx5: Enable MIGRATION_PRE_COPY flag

Now that everything has been set up for MIGRATION_PRE_COPY, enable it.

Signed-off-by: Shay Drory <shayd@nvidia.com>
Reviewed-by: Jason Gunthorpe <jgg@nvidia.com>
Signed-off-by: Yishai Hadas <yishaih@nvidia.com>
Link: https://lore.kernel.org/r/20221206083438.37807-15-yishaih@nvidia.com
Signed-off-by: Alex Williamson <alex.williamson@redhat.com>

diff --git a/drivers/vfio/pci/mlx5/cmd.c b/drivers/vfio/pci/mlx5/cmd.c
index 01ef695e94418de246fc729b00cef4e99e4b6e26..64e68d13cb98fc1812b8f096cc23c80ef2919bc2 100644 (file)
--- a/drivers/vfio/pci/mlx5/cmd.c
+++ b/drivers/vfio/pci/mlx5/cmd.c
@@ -222,6 +222,11 @@ void mlx5vf_cmd_set_migratable(struct mlx5vf_pci_core_device *mvdev,
        if (MLX5_CAP_GEN(mvdev->mdev, adv_virtualization))
                mvdev->core_device.vdev.log_ops = log_ops;
 
+       if (MLX5_CAP_GEN_2(mvdev->mdev, migration_multi_load) &&
+           MLX5_CAP_GEN_2(mvdev->mdev, migration_tracking_state))
+               mvdev->core_device.vdev.migration_flags |=
+                       VFIO_MIGRATION_PRE_COPY;
+
 end:
        mlx5_vf_put_core_dev(mvdev->mdev);
 }