Merge branch 'drm-next' of git://people.freedesktop.org/~airlied/linux

[sagit-ice-cold/kernel_xiaomi_msm8998.git] / virt / kvm / iommu.c

diff --git a/virt/kvm/iommu.c b/virt/kvm/iommu.c
index 714b949..e723bb9 100644 (file)
--- a/virt/kvm/iommu.c
+++ b/virt/kvm/iommu.c
@@ -203,7 +203,7 @@ int kvm_assign_device(struct kvm *kvm,
                        goto out_unmap;
        }
 
-       pdev->dev_flags |= PCI_DEV_FLAGS_ASSIGNED;
+       pci_set_dev_assigned(pdev);
 
        dev_info(&pdev->dev, "kvm assign device\n");
 
@@ -229,7 +229,7 @@ int kvm_deassign_device(struct kvm *kvm,
 
        iommu_detach_device(domain, &pdev->dev);
 
-       pdev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
+       pci_clear_dev_assigned(pdev);
 
        dev_info(&pdev->dev, "kvm deassign device\n");