drm/xe/tests: Support CPU page-table updates in the migrate test

The migrate test currently supports only GPU pagetable updates and
will thus break if we fix the CPU pagetable update selection.

Fix the migrate test first.

Signed-off-by: Thomas Hellström <thomas.hellstrom@linux.intel.com>
Reviewed-by: Matthew Auld <matthew.auld@intel.com>
Signed-off-by: Rodrigo Vivi <rodrigo.vivi@intel.com>

diff --git a/drivers/gpu/drm/xe/tests/xe_migrate.c b/drivers/gpu/drm/xe/tests/xe_migrate.c
index ac659b94e7f5237bebb3d6972e52b0df40982f3e..a3bace16282ee80da4c3b9f39ec0a52f73c3ace9 100644 (file)
--- a/drivers/gpu/drm/xe/tests/xe_migrate.c
+++ b/drivers/gpu/drm/xe/tests/xe_migrate.c
@@ -66,9 +66,16 @@ sanity_populate_cb(struct xe_migrate_pt_update *pt_update,
 {
        int i;
        u64 *ptr = dst;
-
-       for (i = 0; i < num_qwords; i++)
-               ptr[i] = (qword_ofs + i - update->ofs) * 0x1111111111111111ULL;
+       u64 value;
+
+       for (i = 0; i < num_qwords; i++) {
+               value = (qword_ofs + i - update->ofs) * 0x1111111111111111ULL;
+               if (map)
+                       xe_map_wr(gt_to_xe(gt), map, (qword_ofs + i) *
+                                 sizeof(u64), u64, value);
+               else
+                       ptr[i] = value;
+       }
 }
 
 static const struct xe_migrate_pt_update_ops sanity_ops = {