[PATCH 4/5] testme3

[Maarten Lankhorst]

Signed-off-by: Maarten Lankhorst <maarten.lankhorst at linux.intel.com>
---
 drivers/gpu/drm/i915/gt/intel_ggtt.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/drivers/gpu/drm/i915/gt/intel_ggtt.c b/drivers/gpu/drm/i915/gt/intel_ggtt.c
index 5fd66f4ff6b0..bd8e7dea6254 100644
--- a/drivers/gpu/drm/i915/gt/intel_ggtt.c
+++ b/drivers/gpu/drm/i915/gt/intel_ggtt.c
@@ -370,7 +370,9 @@ static void bxt_vtd_ggtt_insert_page__BKL(struct i915_address_space *vm,
 {
 	struct insert_page arg = { vm, addr, offset, level };
 
-	stop_machine(bxt_vtd_ggtt_insert_page__cb, &arg, NULL);
+	migrate_disable();
+	stop_machine(bxt_vtd_ggtt_insert_page__cb, &arg, cpumask_of(smp_processor_id()));
+	migrate_enable();
 }
 
 struct insert_entries {
@@ -397,7 +399,9 @@ static void bxt_vtd_ggtt_insert_entries__BKL(struct i915_address_space *vm,
 {
 	struct insert_entries arg = { vm, vma, level, flags };
 
-	stop_machine(bxt_vtd_ggtt_insert_entries__cb, &arg, NULL);
+	migrate_disable();
+	stop_machine(bxt_vtd_ggtt_insert_entries__cb, &arg, cpumask_of(smp_processor_id()));
+	migrate_enable();
 }
 
 static void gen6_ggtt_clear_range(struct i915_address_space *vm,
-- 
2.31.0