X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=virt%2Fkvm%2Fiommu.c;h=d149940608da17cfd748691f0b8427a4a60f78dd;hb=a1b60c1cd913c5ccfb38c717ba0bd22622425fa7;hp=78c80f67f535f97055d312aa946c109bcfdcd424;hpb=6d158f3ec537bd6d3786196b6db97fcc0f973aef;p=firefly-linux-kernel-4.4.55.git

diff --git a/virt/kvm/iommu.c b/virt/kvm/iommu.c
index 78c80f67f535..d149940608da 100644
--- a/virt/kvm/iommu.c
+++ b/virt/kvm/iommu.c
@@ -228,12 +228,12 @@ int kvm_iommu_map_guest(struct kvm *kvm)
 {
 	int r;
 
-	if (!iommu_found()) {
+	if (!iommu_present(&pci_bus_type)) {
 		printk(KERN_ERR "%s: iommu not found\n", __func__);
 		return -ENODEV;
 	}
 
-	kvm->arch.iommu_domain = iommu_domain_alloc();
+	kvm->arch.iommu_domain = iommu_domain_alloc(&pci_bus_type);
 	if (!kvm->arch.iommu_domain)
 		return -ENOMEM;