1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/iommu/amd_iommu.c b/drivers/iommu/amd_iommu.c
index 0e4227f457af..4ee277a8521a 100644
--- a/drivers/iommu/amd_iommu.c
+++ b/drivers/iommu/amd_iommu.c
@@ -1283,7 +1283,7 @@ static int alloc_new_range(struct dma_ops_domain *dma_dom,
                 if (!pte || !IOMMU_PTE_PRESENT(*pte))
                         continue;
 
-                dma_ops_reserve_addresses(dma_dom, i << PAGE_SHIFT, 1);
+                dma_ops_reserve_addresses(dma_dom, i >> PAGE_SHIFT, 1);
         }
 
         update_domain(&dma_dom->domain);
@@ -2495,7 +2495,7 @@ static unsigned device_dma_ops_init(void)
 
 void __init amd_iommu_init_api(void)
 {
-        register_iommu(&amd_iommu_ops);
+        bus_set_iommu(&pci_bus_type, &amd_iommu_ops);
 }
 
 int __init amd_iommu_init_dma_ops(void)