drm/ttm: add page order support in ttm_pages_put

Reviewed-by: Chunming Zhou <david1.zhou@amd.com>
Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Roger He <Hongbo.He@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 18 insertions, 8 deletions

diff --git a/drivers/gpu/drm/ttm/ttm_page_alloc.c b/drivers/gpu/drm/ttm/ttm_page_alloc.c
index 359a641d3211..337c228b44ad 100644
--- a/drivers/gpu/drm/ttm/ttm_page_alloc.c
+++ b/drivers/gpu/drm/ttm/ttm_page_alloc.c
@@ -296,13 +296,23 @@ static struct ttm_page_pool *ttm_get_pool(int flags, bool huge,
 }
 
 /* set memory back to wb and free the pages. */
-static void ttm_pages_put(struct page *pages[], unsigned npages)
+static void ttm_pages_put(struct page *pages[], unsigned npages,
+                unsigned int order)
 {
-        unsigned i;
-        if (set_pages_array_wb(pages, npages))
-                pr_err("Failed to set %d pages to wb!\n", npages);
-        for (i = 0; i < npages; ++i)
-                __free_page(pages[i]);
+        unsigned int i, pages_nr = (1 << order);
+
+        if (order == 0) {
+                if (set_pages_array_wb(pages, npages))
+                        pr_err("Failed to set %d pages to wb!\n", npages);
+        }
+
+        for (i = 0; i < npages; ++i) {
+                if (order > 0) {
+                        if (set_pages_wb(pages[i], pages_nr))
+                                pr_err("Failed to set %d pages to wb!\n", pages_nr);
+                }
+                __free_pages(pages[i], order);
+        }
 }
 
 static void ttm_pool_update_free_locked(struct ttm_page_pool *pool,
@@ -365,7 +375,7 @@ restart:
                          */
                         spin_unlock_irqrestore(&pool->lock, irq_flags);
 
-                        ttm_pages_put(pages_to_free, freed_pages);
+                        ttm_pages_put(pages_to_free, freed_pages, pool->order);
                         if (likely(nr_free != FREE_ALL_PAGES))
                                 nr_free -= freed_pages;
 
@@ -400,7 +410,7 @@ restart:
         spin_unlock_irqrestore(&pool->lock, irq_flags);
 
         if (freed_pages)
-                ttm_pages_put(pages_to_free, freed_pages);
+                ttm_pages_put(pages_to_free, freed_pages, pool->order);
 out:
         if (pages_to_free != static_buf)
                 kfree(pages_to_free);