Implemented a new intersect and compatible callback function
fetching start offset from drm buddy allocator.

Signed-off-by: Christian König <christian.koe...@amd.com>
Signed-off-by: Arunpravin Paneer Selvam <arunpravin.paneersel...@amd.com>
---
 drivers/gpu/drm/i915/i915_ttm_buddy_manager.c | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/drivers/gpu/drm/i915/i915_ttm_buddy_manager.c 
b/drivers/gpu/drm/i915/i915_ttm_buddy_manager.c
index a5109548abc0..b5801c05bd41 100644
--- a/drivers/gpu/drm/i915/i915_ttm_buddy_manager.c
+++ b/drivers/gpu/drm/i915/i915_ttm_buddy_manager.c
@@ -178,6 +178,47 @@ static void i915_ttm_buddy_man_free(struct 
ttm_resource_manager *man,
        kfree(bman_res);
 }
 
+static bool i915_ttm_buddy_man_intersect(struct ttm_resource_manager *man,
+                                        struct ttm_resource *res,
+                                        const struct ttm_place *place,
+                                        size_t size)
+{
+       struct i915_ttm_buddy_resource *bman_res = to_ttm_buddy_resource(res);
+       u32 start, num_pages = PFN_UP(size);
+       struct drm_buddy_block *block;
+
+       /* Check each drm buddy block individually */
+       list_for_each_entry(block, &bman_res->blocks, link) {
+               start = drm_buddy_block_offset(block) >> PAGE_SHIFT;
+               /* Don't evict BOs outside of the requested placement range */
+               if (place->fpfn >= (start + num_pages) ||
+                   (place->lpfn && place->lpfn <= start))
+                       return false;
+       }
+
+       return true;
+}
+
+static bool i915_ttm_buddy_man_compatible(struct ttm_resource_manager *man,
+                                         struct ttm_resource *res,
+                                         const struct ttm_place *place,
+                                         size_t size)
+{
+       struct i915_ttm_buddy_resource *bman_res = to_ttm_buddy_resource(res);
+       u32 start, num_pages = PFN_UP(size);
+       struct drm_buddy_block *block;
+
+       /* Check each drm buddy block individually */
+       list_for_each_entry(block, &bman_res->blocks, link) {
+               start = drm_buddy_block_offset(block) >> PAGE_SHIFT;
+               if (start < place->fpfn ||
+                   (place->lpfn && (start + num_pages) > place->lpfn))
+                       return false;
+       }
+
+       return true;
+}
+
 static void i915_ttm_buddy_man_debug(struct ttm_resource_manager *man,
                                     struct drm_printer *printer)
 {
@@ -205,6 +246,8 @@ static void i915_ttm_buddy_man_debug(struct 
ttm_resource_manager *man,
 static const struct ttm_resource_manager_func i915_ttm_buddy_manager_func = {
        .alloc = i915_ttm_buddy_man_alloc,
        .free = i915_ttm_buddy_man_free,
+       .intersect = i915_ttm_buddy_man_intersect,
+       .compatible = i915_ttm_buddy_man_compatible,
        .debug = i915_ttm_buddy_man_debug,
 };
 
-- 
2.25.1

Reply via email to