[i915-rt v4 17/20] drm/i915/gt: Fix selftests on PREEMPT_RT

[Maarten Lankhorst]

Signed-off-by: Maarten Lankhorst <d...@lankhorst.se>
---
 drivers/gpu/drm/i915/gt/selftest_engine_pm.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/drivers/gpu/drm/i915/gt/selftest_engine_pm.c 
b/drivers/gpu/drm/i915/gt/selftest_engine_pm.c
index 10e556a7eac45..c1eff9edd8a5e 100644
--- a/drivers/gpu/drm/i915/gt/selftest_engine_pm.c
+++ b/drivers/gpu/drm/i915/gt/selftest_engine_pm.c
@@ -277,11 +277,11 @@ static int live_engine_busy_stats(void *arg)
                st_engine_heartbeat_disable(engine);
 
                ENGINE_TRACE(engine, "measuring idle time\n");
-               preempt_disable();
+               migrate_disable();
                de = intel_engine_get_busy_time(engine, &t[0]);
                udelay(100);
                de = ktime_sub(intel_engine_get_busy_time(engine, &t[1]), de);
-               preempt_enable();
+               migrate_enable();
                dt = ktime_sub(t[1], t[0]);
                if (de < 0 || de > 10) {
                        pr_err("%s: reported %lldns [%d%%] busyness while 
sleeping [for %lldns]\n",
@@ -316,11 +316,11 @@ static int live_engine_busy_stats(void *arg)
                }
 
                ENGINE_TRACE(engine, "measuring busy time\n");
-               preempt_disable();
+               migrate_disable();
                de = intel_engine_get_busy_time(engine, &t[0]);
                mdelay(100);
                de = ktime_sub(intel_engine_get_busy_time(engine, &t[1]), de);
-               preempt_enable();
+               migrate_enable();
                dt = ktime_sub(t[1], t[0]);
                if (100 * de < 95 * dt || 95 * de > 100 * dt) {
                        pr_err("%s: reported %lldns [%d%%] busyness while 
spinning [for %lldns]\n",
-- 
2.51.0