drm/i915/selftests: Return to kernel context after each test

As we flush each test and wait for idle before the next, also switch
back to the kernel context. This helps limit the amount of collateral
damage a test may cause by resetting to the default state each time (and
also helps clean up temporaries used by the test).

Signed-off-by: Chris Wilson <[email protected]>
Cc: Mika Kuoppala <[email protected]>
Reviewed-by: Mika Kuoppala <[email protected]>
Link: https://patchwork.freedesktop.org/patch/msgid/[email protected]

diff --git a/drivers/gpu/drm/i915/selftests/igt_flush_test.c b/drivers/gpu/drm/i915/selftests/igt_flush_test.c
index abff2f04ea84a11f4610fd93f876cfbabdcd63ac..7f35bddc2e9522ca8cb70a80374df8dc65f4baf9 100644 (file)
--- a/drivers/gpu/drm/i915/selftests/igt_flush_test.c
+++ b/drivers/gpu/drm/i915/selftests/igt_flush_test.c
@@ -57,6 +57,11 @@ int igt_flush_test(struct drm_i915_private *i915, unsigned int flags)
 
        cond_resched();
 
+       if (i915_gem_switch_to_kernel_context(i915)) {
+               pr_err("Failed to switch back to kernel context; declaring wedged\n");
+               i915_gem_set_wedged(i915);
+       }
+
        wedge_on_timeout(&w, i915, HZ)
                i915_gem_wait_for_idle(i915, flags);