Mock test filling an address space with 4K and 64K objects, in the hope
of exercising the page color adjust fun.

Signed-off-by: Matthew Auld <matthew.a...@intel.com>
---
 drivers/gpu/drm/i915/selftests/i915_gem_gtt.c | 68 +++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c 
b/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
index c7963efe46ba..9b2a7228a78f 100644
--- a/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
+++ b/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
@@ -31,6 +31,7 @@
 #include "mock_context.h"
 #include "mock_drm.h"
 #include "mock_gem_device.h"
+#include "mock_gtt.h"
 
 static void fake_free_pages(struct drm_i915_gem_object *obj,
                            struct sg_table *pages)
@@ -1306,6 +1307,72 @@ static int igt_gtt_reserve(void *arg)
        return err;
 }
 
+static int igt_ppgtt_page_color(void *arg)
+{
+       struct drm_mm mm;
+       struct drm_mm_node *node, *prev, *next;
+       unsigned long page_colors[] = {
+               I915_GTT_PAGE_SIZE,
+               I915_GTT_PAGE_SIZE_64K,
+       };
+       int idx = 0;
+       u64 count = 0;
+       u64 size;
+
+       drm_mm_init(&mm, 0, U64_MAX);
+       mm.color_adjust = i915_page_color_adjust;
+
+       /* Running out of memory is okay. */
+
+       for_each_prime_number_from(size, 0, U64_MAX) {
+               node = kzalloc(sizeof(*node), GFP_KERNEL);
+               if (!node) {
+                       pr_info("finished test early, unable to allocate node, 
count=%llu\n", count);
+                       break;
+               }
+
+               size = roundup(size, page_colors[idx]);
+
+               if (drm_mm_insert_node_in_range(&mm, node, size,
+                                               page_colors[idx],
+                                               page_colors[idx],
+                                               0, U64_MAX,
+                                               DRM_MM_INSERT_BEST)) {
+                       pr_info("test finished, unable to insert node: 
color=%lu, size=%llx, count=%llu\n",
+                               page_colors[idx], size, count);
+                       kfree(node);
+                       break;
+               }
+
+               GEM_BUG_ON(!IS_ALIGNED(node->start, node->color));
+               GEM_BUG_ON(!IS_ALIGNED(node->size, node->color));
+
+               /* We can't mix 4K and 64K pte's in the same pt. */
+
+               prev = list_prev_entry(node, node_list);
+               if (i915_color_differs(prev, node->color))
+                       GEM_BUG_ON(prev->start >> GEN8_PDE_SHIFT ==
+                                  node->start >> GEN8_PDE_SHIFT);
+
+               next = list_next_entry(node, node_list);
+               if (i915_color_differs(next, node->color))
+                       GEM_BUG_ON(((next->start + next->size) >> 
GEN8_PDE_SHIFT) ==
+                                  ((node->start + node->size) >> 
GEN8_PDE_SHIFT));
+
+               idx ^= 1;
+               ++count;
+       }
+
+       drm_mm_for_each_node_safe(node, next, &mm) {
+               drm_mm_remove_node(node);
+               kfree(node);
+       }
+
+       drm_mm_takedown(&mm);
+
+       return 0;
+}
+
 static int igt_gtt_insert(void *arg)
 {
        struct drm_i915_private *i915 = arg;
@@ -1522,6 +1589,7 @@ int i915_gem_gtt_mock_selftests(void)
                SUBTEST(igt_mock_fill),
                SUBTEST(igt_gtt_reserve),
                SUBTEST(igt_gtt_insert),
+               SUBTEST(igt_ppgtt_page_color),
        };
        struct drm_i915_private *i915;
        int err;
-- 
2.9.3

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

Reply via email to