iommu/pages: use folio_nr_pages() instead of shift operation

folio_nr_pages() is a faster helper function to get the number of pages when
NR_PAGES_IN_LARGE_FOLIO is enabled.

Signed-off-by: Pedro Demarchi Gomes <pedrodemargomes@gmail.com>
Reviewed-by: Jason Gunthorpe <jgg@nvidia.com>
Signed-off-by: Joerg Roedel <joerg.roedel@amd.com>

diff --git a/drivers/iommu/iommu-pages.c b/drivers/iommu/iommu-pages.c
index 238c09e5166b4d33f09c7b044b59e32c72cb7d50..3cb8fcc390660257f61964e9c93bfc503e441150 100644 (file)
--- a/drivers/iommu/iommu-pages.c
+++ b/drivers/iommu/iommu-pages.c
@@ -80,7 +80,7 @@ EXPORT_SYMBOL_GPL(iommu_alloc_pages_node_sz);
 static void __iommu_free_desc(struct ioptdesc *iopt)
 {
        struct folio *folio = ioptdesc_folio(iopt);
-       const unsigned long pgcnt = 1UL << folio_order(folio);
+       const unsigned long pgcnt = folio_nr_pages(folio);
 
        mod_node_page_state(folio_pgdat(folio), NR_IOMMU_PAGES, -pgcnt);
        lruvec_stat_mod_folio(folio, NR_SECONDARY_PAGETABLE, -pgcnt);