1 files changed, 4 insertions, 3 deletions

diff --git a/arch/sparc64/kernel/iommu.c b/arch/sparc64/kernel/iommu.c
index 2a37a6ca2a16..1cc1995531e2 100644
--- a/arch/sparc64/kernel/iommu.c
+++ b/arch/sparc64/kernel/iommu.c
@@ -575,7 +575,7 @@ static int dma_4u_map_sg(struct device *dev, struct scatterlist *sglist,
                 }
                 /* Allocate iommu entries for that segment */
                 paddr = (unsigned long) SG_ENT_PHYS_ADDRESS(s);
-                npages = iommu_num_pages(paddr, slen);
+                npages = iommu_num_pages(paddr, slen, IO_PAGE_SIZE);
                 entry = iommu_range_alloc(dev, iommu, npages, &handle);
 
                 /* Handle failure */
@@ -647,7 +647,8 @@ iommu_map_failed:
                         iopte_t *base;
 
                         vaddr = s->dma_address & IO_PAGE_MASK;
-                        npages = iommu_num_pages(s->dma_address, s->dma_length);
+                        npages = iommu_num_pages(s->dma_address, s->dma_length,
+                                                 IO_PAGE_SIZE);
                         iommu_range_free(iommu, vaddr, npages);
 
                         entry = (vaddr - iommu->page_table_map_base)
@@ -715,7 +716,7 @@ static void dma_4u_unmap_sg(struct device *dev, struct scatterlist *sglist,
 
                 if (!len)
                         break;
-                npages = iommu_num_pages(dma_handle, len);
+                npages = iommu_num_pages(dma_handle, len, IO_PAGE_SIZE);
                 iommu_range_free(iommu, dma_handle, npages);
 
                 entry = ((dma_handle - iommu->page_table_map_base)