1 files changed, 4 insertions, 4 deletions

diff --git a/virt/kvm/iommu.c b/virt/kvm/iommu.c
index e51d9f9b995f..c1e6ae989a43 100644
--- a/virt/kvm/iommu.c
+++ b/virt/kvm/iommu.c
@@ -43,13 +43,13 @@ static void kvm_iommu_put_pages(struct kvm *kvm,
                                 gfn_t base_gfn, unsigned long npages);
 
 static pfn_t kvm_pin_pages(struct kvm_memory_slot *slot, gfn_t gfn,
-                           unsigned long size)
+                           unsigned long npages)
 {
         gfn_t end_gfn;
         pfn_t pfn;
 
         pfn     = gfn_to_pfn_memslot(slot, gfn);
-        end_gfn = gfn + (size >> PAGE_SHIFT);
+        end_gfn = gfn + npages;
         gfn    += 1;
 
         if (is_error_noslot_pfn(pfn))
@@ -119,7 +119,7 @@ int kvm_iommu_map_pages(struct kvm *kvm, struct kvm_memory_slot *slot)
                  * Pin all pages we are about to map in memory. This is
                  * important because we unmap and unpin in 4kb steps later.
                  */
-                pfn = kvm_pin_pages(slot, gfn, page_size);
+                pfn = kvm_pin_pages(slot, gfn, page_size >> PAGE_SHIFT);
                 if (is_error_noslot_pfn(pfn)) {
                         gfn += 1;
                         continue;
@@ -131,7 +131,7 @@ int kvm_iommu_map_pages(struct kvm *kvm, struct kvm_memory_slot *slot)
                 if (r) {
                         printk(KERN_ERR "kvm_iommu_map_address:"
                                "iommu failed to map pfn=%llx\n", pfn);
-                        kvm_unpin_pages(kvm, pfn, page_size);
+                        kvm_unpin_pages(kvm, pfn, page_size >> PAGE_SHIFT);
                         goto unmap_pages;
                 }