drm/xe/tests: Use proper batch base address

In xe_migrate_sanity_kunit test, use proper batch base address
by considering usm case.

Reviewed-by: Matthew Brost <matthew.brost@intel.com>
Signed-off-by: Niranjana Vishwanathapura <niranjana.vishwanathapura@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 17829f878757ec0b5cd8c39f05e4e5c01bd6ca7e..90f4e1c4f029f0e185e1f48883cdb289db5588e9 100644 (file)
--- a/drivers/gpu/drm/xe/tests/xe_migrate.c
+++ b/drivers/gpu/drm/xe/tests/xe_migrate.c
@@ -35,8 +35,9 @@ static int run_sanity_job(struct xe_migrate *m, struct xe_device *xe,
                          struct xe_bb *bb, u32 second_idx, const char *str,
                          struct kunit *test)
 {
+       u64 batch_base = xe_migrate_batch_base(m, xe->info.supports_usm);
        struct xe_sched_job *job = xe_bb_create_migration_job(m->eng, bb,
-                                                             m->batch_base_ofs,
+                                                             batch_base,
                                                              second_idx);
        struct dma_fence *fence;