2 files changed, 4 insertions, 0 deletions

diff --git a/arch/ia64/kvm/kvm-ia64.c b/arch/ia64/kvm/kvm-ia64.c
index dcc560729c20..5b2dc0d10c8f 100644
--- a/arch/ia64/kvm/kvm-ia64.c
+++ b/arch/ia64/kvm/kvm-ia64.c
@@ -204,9 +204,11 @@ int kvm_dev_ioctl_check_extension(long ext)
         case KVM_CAP_COALESCED_MMIO:
                 r = KVM_COALESCED_MMIO_PAGE_OFFSET;
                 break;
+#ifdef CONFIG_KVM_DEVICE_ASSIGNMENT
         case KVM_CAP_IOMMU:
                 r = iommu_present(&pci_bus_type);
                 break;
+#endif
         default:
                 r = 0;
         }
diff --git a/arch/x86/kvm/x86.c b/arch/x86/kvm/x86.c
index 8747fef7fd59..96f914e828d4 100644
--- a/arch/x86/kvm/x86.c
+++ b/arch/x86/kvm/x86.c
@@ -2544,9 +2544,11 @@ int kvm_dev_ioctl_check_extension(long ext)
         case KVM_CAP_PV_MMU:    /* obsolete */
                 r = 0;
                 break;
+#ifdef CONFIG_KVM_DEVICE_ASSIGNMENT
         case KVM_CAP_IOMMU:
                 r = iommu_present(&pci_bus_type);
                 break;
+#endif
         case KVM_CAP_MCE:
                 r = KVM_MAX_MCE_BANKS;
                 break;