}
 
 static int sun50i_iommu_map(struct iommu_domain *domain, unsigned long iova,
-                           phys_addr_t paddr, size_t size, int prot, gfp_t gfp)
+                           phys_addr_t paddr, size_t size, size_t count,
+                           int prot, gfp_t gfp, size_t *mapped)
 {
        struct sun50i_iommu_domain *sun50i_domain = to_sun50i_domain(domain);
        struct sun50i_iommu *iommu = sun50i_domain->iommu;
 
        *pte_addr = sun50i_mk_pte(paddr, prot);
        sun50i_table_flush(sun50i_domain, pte_addr, 1);
+       *mapped = size;
 
 out:
        return ret;
 }
 
 static size_t sun50i_iommu_unmap(struct iommu_domain *domain, unsigned long iova,
-                                size_t size, struct iommu_iotlb_gather *gather)
+                                size_t size, size_t count, struct iommu_iotlb_gather *gather)
 {
        struct sun50i_iommu_domain *sun50i_domain = to_sun50i_domain(domain);
        phys_addr_t pt_phys;
                .iotlb_sync_map = sun50i_iommu_iotlb_sync_map,
                .iotlb_sync     = sun50i_iommu_iotlb_sync,
                .iova_to_phys   = sun50i_iommu_iova_to_phys,
-               .map            = sun50i_iommu_map,
-               .unmap          = sun50i_iommu_unmap,
+               .map_pages      = sun50i_iommu_map,
+               .unmap_pages    = sun50i_iommu_unmap,
                .free           = sun50i_iommu_domain_free,
        }
 };