prepare for ttm based stolen region by using ttm range manager
as the resource manager for stolen region.

Signed-off-by: Robert Beckett <bob.beck...@collabora.com>
Reviewed-by: Thomas Hellström <thomas.hellst...@linux.intel.com>
---
 drivers/gpu/drm/i915/gem/i915_gem_ttm_move.c |  6 ++--
 drivers/gpu/drm/i915/intel_region_ttm.c      | 31 +++++++++++++++-----
 2 files changed, 27 insertions(+), 10 deletions(-)

diff --git a/drivers/gpu/drm/i915/gem/i915_gem_ttm_move.c 
b/drivers/gpu/drm/i915/gem/i915_gem_ttm_move.c
index 40249fa28a7a..675e9ab30396 100644
--- a/drivers/gpu/drm/i915/gem/i915_gem_ttm_move.c
+++ b/drivers/gpu/drm/i915/gem/i915_gem_ttm_move.c
@@ -60,11 +60,13 @@ i915_ttm_region(struct ttm_device *bdev, int ttm_mem_type)
        struct drm_i915_private *i915 = container_of(bdev, typeof(*i915), bdev);
 
        /* There's some room for optimization here... */
-       GEM_BUG_ON(ttm_mem_type != I915_PL_SYSTEM &&
-                  ttm_mem_type < I915_PL_LMEM0);
+       GEM_BUG_ON(ttm_mem_type == I915_PL_GGTT);
+
        if (ttm_mem_type == I915_PL_SYSTEM)
                return intel_memory_region_lookup(i915, INTEL_MEMORY_SYSTEM,
                                                  0);
+       if (ttm_mem_type == I915_PL_STOLEN)
+               return i915->mm.stolen_region;
 
        return intel_memory_region_lookup(i915, INTEL_MEMORY_LOCAL,
                                          ttm_mem_type - I915_PL_LMEM0);
diff --git a/drivers/gpu/drm/i915/intel_region_ttm.c 
b/drivers/gpu/drm/i915/intel_region_ttm.c
index fd2ecfdd8fa1..694e9acb69e2 100644
--- a/drivers/gpu/drm/i915/intel_region_ttm.c
+++ b/drivers/gpu/drm/i915/intel_region_ttm.c
@@ -54,7 +54,7 @@ void intel_region_ttm_device_fini(struct drm_i915_private 
*dev_priv)
 
 /*
  * Map the i915 memory regions to TTM memory types. We use the
- * driver-private types for now, reserving TTM_PL_VRAM for stolen
+ * driver-private types for now, reserving I915_PL_STOLEN for stolen
  * memory and TTM_PL_TT for GGTT use if decided to implement this.
  */
 int intel_region_to_ttm_type(const struct intel_memory_region *mem)
@@ -63,11 +63,17 @@ int intel_region_to_ttm_type(const struct 
intel_memory_region *mem)
 
        GEM_BUG_ON(mem->type != INTEL_MEMORY_LOCAL &&
                   mem->type != INTEL_MEMORY_MOCK &&
-                  mem->type != INTEL_MEMORY_SYSTEM);
+                  mem->type != INTEL_MEMORY_SYSTEM &&
+                  mem->type != INTEL_MEMORY_STOLEN_SYSTEM &&
+                  mem->type != INTEL_MEMORY_STOLEN_LOCAL);
 
        if (mem->type == INTEL_MEMORY_SYSTEM)
                return TTM_PL_SYSTEM;
 
+       if (mem->type == INTEL_MEMORY_STOLEN_SYSTEM ||
+           mem->type == INTEL_MEMORY_STOLEN_LOCAL)
+               return I915_PL_STOLEN;
+
        type = mem->instance + TTM_PL_PRIV;
        GEM_BUG_ON(type >= TTM_NUM_MEM_TYPES);
 
@@ -91,10 +97,16 @@ int intel_region_ttm_init(struct intel_memory_region *mem)
        int mem_type = intel_region_to_ttm_type(mem);
        int ret;
 
-       ret = i915_ttm_buddy_man_init(bdev, mem_type, false,
-                                     resource_size(&mem->region),
-                                     mem->io_size,
-                                     mem->min_page_size, PAGE_SIZE);
+       if (mem_type == I915_PL_STOLEN) {
+               ret = ttm_range_man_init(bdev, mem_type, false,
+                                        resource_size(&mem->region) >> 
PAGE_SHIFT);
+               mem->is_range_manager = true;
+       } else {
+               ret = i915_ttm_buddy_man_init(bdev, mem_type, false,
+                                             resource_size(&mem->region),
+                                             mem->io_size,
+                                             mem->min_page_size, PAGE_SIZE);
+       }
        if (ret)
                return ret;
 
@@ -114,6 +126,7 @@ int intel_region_ttm_init(struct intel_memory_region *mem)
 int intel_region_ttm_fini(struct intel_memory_region *mem)
 {
        struct ttm_resource_manager *man = mem->region_private;
+       int mem_type = intel_region_to_ttm_type(mem);
        int ret = -EBUSY;
        int count;
 
@@ -144,8 +157,10 @@ int intel_region_ttm_fini(struct intel_memory_region *mem)
        if (ret || !man)
                return ret;
 
-       ret = i915_ttm_buddy_man_fini(&mem->i915->bdev,
-                                     intel_region_to_ttm_type(mem));
+       if (mem_type == I915_PL_STOLEN)
+               ret = ttm_range_man_fini(&mem->i915->bdev, mem_type);
+       else
+               ret = i915_ttm_buddy_man_fini(&mem->i915->bdev, mem_type);
        GEM_WARN_ON(ret);
        mem->region_private = NULL;
 
-- 
2.25.1

Reply via email to