Separate the drm_mm_node walking from the actual handling.

Signed-off-by: Christian König <christian.koe...@amd.com>
Acked-by: Oak Zeng <oak.z...@amd.com>
Tested-by: Nirmoy Das <nirmoy....@amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
index d469ba5fef2c..5d88d1850781 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_ttm.c
@@ -1398,7 +1398,7 @@ static bool amdgpu_ttm_bo_eviction_valuable(struct 
ttm_buffer_object *bo,
                                            const struct ttm_place *place)
 {
        unsigned long num_pages = bo->mem.num_pages;
-       struct drm_mm_node *node = bo->mem.mm_node;
+       struct amdgpu_res_cursor cursor;
        struct dma_resv_list *flist;
        struct dma_fence *f;
        int i;
@@ -1430,13 +1430,15 @@ static bool amdgpu_ttm_bo_eviction_valuable(struct 
ttm_buffer_object *bo,
 
        case TTM_PL_VRAM:
                /* Check each drm MM node individually */
-               while (num_pages) {
-                       if (place->fpfn < (node->start + node->size) &&
-                           !(place->lpfn && place->lpfn <= node->start))
+               amdgpu_res_first(&bo->mem, 0, (u64)num_pages << PAGE_SHIFT,
+                                &cursor);
+               while (cursor.remaining) {
+                       if (place->fpfn < PFN_DOWN(cursor.start + cursor.size)
+                           && !(place->lpfn &&
+                                place->lpfn <= PFN_DOWN(cursor.start)))
                                return true;
 
-                       num_pages -= node->size;
-                       ++node;
+                       amdgpu_res_next(&cursor, cursor.size);
                }
                return false;
 
-- 
2.25.1

_______________________________________________
amd-gfx mailing list
amd-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to