Limit the GTT size we try and allocate to ensure that it fits within RAM
and does not trigger the oomkiller indiscriminately.

Signed-off-by: Chris Wilson <[email protected]>
Cc: Matthew Auld <[email protected]>
---
 drivers/gpu/drm/i915/selftests/i915_gem_gtt.c | 20 ++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c 
b/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
index e108fe4e0fd9..600a3bcbd3d6 100644
--- a/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
+++ b/drivers/gpu/drm/i915/selftests/i915_gem_gtt.c
@@ -148,7 +148,7 @@ static int igt_ppgtt_alloc(void *arg)
 {
        struct drm_i915_private *dev_priv = arg;
        struct i915_hw_ppgtt *ppgtt;
-       u64 size, last;
+       u64 size, last, limit;
        int err = 0;
 
        /* Allocate a ppggt and try to fill the entire range */
@@ -163,10 +163,18 @@ static int igt_ppgtt_alloc(void *arg)
        if (!ppgtt->vm.allocate_va_range)
                goto err_ppgtt_cleanup;
 
+       /*
+        * While we only allocate the page tables here and so we could
+        * address a much larger GTT than we could actually fit into
+        * RAM, a practical limit is the amount of physical pages in the system.
+        * This should ensure that we do not run into the oomkiller during
+        * the test and take down the machine wilfully.
+        */
+       limit = totalram_pages << PAGE_SHIFT;
+       limit = min(ppgtt->vm.total, limit);
+
        /* Check we can allocate the entire range */
-       for (size = 4096;
-            size <= ppgtt->vm.total;
-            size <<= 2) {
+       for (size = 4096; size <= limit; size <<= 2) {
                err = ppgtt->vm.allocate_va_range(&ppgtt->vm, 0, size);
                if (err) {
                        if (err == -ENOMEM) {
@@ -183,9 +191,7 @@ static int igt_ppgtt_alloc(void *arg)
        }
 
        /* Check we can incrementally allocate the entire range */
-       for (last = 0, size = 4096;
-            size <= ppgtt->vm.total;
-            last = size, size <<= 2) {
+       for (last = 0, size = 4096; size <= limit; last = size, size <<= 2) {
                err = ppgtt->vm.allocate_va_range(&ppgtt->vm,
                                                  last, size - last);
                if (err) {
-- 
2.18.0

_______________________________________________
Intel-gfx mailing list
[email protected]
https://lists.freedesktop.org/mailman/listinfo/intel-gfx

Reply via email to