[PATCH i-g-t 3/3] tests/intel/xe_pm: Tests vm-unbind all flag functionality with S&R

[sai.gowtham.ch@intel.com]

From: Sai Gowtham Ch <sai.gowtham.ch@intel.com>

Test validates vm unbind all flag functionality with suspend and resume.

Cc: Rodrigo Vivi <rodrigo.vivi@intel.com>
Signed-off-by: Sai Gowtham Ch <sai.gowtham.ch@intel.com>
---
 tests/intel/xe_pm.c | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/tests/intel/xe_pm.c b/tests/intel/xe_pm.c
index 839540f5c..6776f5505 100644
--- a/tests/intel/xe_pm.c
+++ b/tests/intel/xe_pm.c
@@ -35,6 +35,7 @@
 
 #define USERPTR         (0x1 << 0)
 #define PREFETCH        (0x1 << 2)
+#define UNBIND_ALL	(0x1 << 3)
 
 typedef struct {
 	int fd_xe;
@@ -219,6 +220,7 @@ static void close_fw_handle(int sig)
 	close(fw_handle);
 }
 
+#define MAX_VMAS 2
 /**
  * SUBTEST: %s-basic
  * Description: set GPU state to %arg[1] and test suspend/autoresume
@@ -291,6 +293,7 @@ static void close_fw_handle(int sig)
  *
  * @usrptr:     usrptr
  * @prefetch:	prefetch
+ * @unbind-all:	unbind-all
  */
 static void
 test_exec(device_t device, struct drm_xe_engine_class_instance *eci,
@@ -308,6 +311,7 @@ test_exec(device_t device, struct drm_xe_engine_class_instance *eci,
 		.num_syncs = 2,
 		.syncs = to_user_pointer(sync),
 	};
+	int n_vmas = flags & UNBIND_ALL ? MAX_VMAS : 1;
 	uint32_t exec_queues[MAX_N_EXEC_QUEUES];
 	uint32_t bind_exec_queues[MAX_N_EXEC_QUEUES];
 	uint32_t syncobjs[MAX_N_EXEC_QUEUES];
@@ -358,8 +362,9 @@ test_exec(device_t device, struct drm_xe_engine_class_instance *eci,
 	sync[0].handle = syncobj_create(device.fd_xe, 0);
 
 	if (bo)
-		xe_vm_bind_async(device.fd_xe, vm, bind_exec_queues[0], bo, 0, addr,
-				 bo_size, sync, 1);
+		for (i = 0; i < n_vmas; i++)
+			xe_vm_bind_async(device.fd_xe, vm, bind_exec_queues[0], bo, 0,
+					 addr + i * bo_size, bo_size, sync, 1);
 	else
 		xe_vm_bind_userptr_async(device.fd_xe, vm, bind_exec_queues[0],
 					 to_user_pointer(data), addr, bo_size, sync, 1);
@@ -414,8 +419,13 @@ test_exec(device_t device, struct drm_xe_engine_class_instance *eci,
 		rpm_usage = igt_pm_get_runtime_usage(device.pci_xe);
 
 	sync[0].flags |= DRM_XE_SYNC_FLAG_SIGNAL;
-	xe_vm_unbind_async(device.fd_xe, vm, bind_exec_queues[0], 0, addr,
-			   bo_size, sync, 1);
+
+	if (n_vmas == MAX_VMAS)
+		xe_vm_unbind_all_async(device.fd_xe, vm, 0, bo, sync, 1);
+	else
+		xe_vm_unbind_async(device.fd_xe, vm, bind_exec_queues[0], 0, addr,
+				   bo_size, sync, 1);
+
 	igt_assert(syncobj_wait(device.fd_xe, &sync[0].handle, 1, INT64_MAX, 0,
 NULL));
 
@@ -624,6 +634,7 @@ igt_main
 	} vm_op[] = {
 		{ "usrptr", USERPTR },
 		{ "prefetch", PREFETCH },
+		{ "unbind-all", UNBIND_ALL },
 		{ NULL },
 	};
 
-- 
2.39.1