[PATCH i-g-t 5/5] tests/intel/xe_sriov_scheduling: Prefer SR-IOV admin sysfs accessors

[Marcin Bernatowicz <marcin.bernatowicz@linux.intel.com>]

Use SR-IOV admin bulk setters for scheduling parameters when present,
falling back to debugfs attributes.

Signed-off-by: Marcin Bernatowicz <marcin.bernatowicz@linux.intel.com>
Cc: Adam Miszczak <adam.miszczak@linux.intel.com>
Cc: Jakub Kolakowski <jakub1.kolakowski@intel.com>
Cc: Lukasz Laguna <lukasz.laguna@intel.com>
Cc: Michal Wajdeczko <michal.wajdeczko@intel.com>
---
 tests/intel/xe_sriov_scheduling.c | 55 ++++++++++++++++++++-----------
 1 file changed, 36 insertions(+), 19 deletions(-)

diff --git a/tests/intel/xe_sriov_scheduling.c b/tests/intel/xe_sriov_scheduling.c
index 3a0c0d438..1e5e19334 100644
--- a/tests/intel/xe_sriov_scheduling.c
+++ b/tests/intel/xe_sriov_scheduling.c
@@ -9,6 +9,7 @@
 #include "xe_drm.h"
 #include "xe/xe_ioctl.h"
 #include "xe/xe_spin.h"
+#include "xe/xe_sriov_admin.h"
 #include "xe/xe_sriov_provisioning.h"
 
 /**
@@ -374,27 +375,43 @@ struct vf_sched_params {
 static int __set_vfs_scheduling_params(int pf_fd, int num_vfs,
 				       const struct vf_sched_params *p)
 {
-	unsigned int gt;
-	int vf, ret;
-
-	xe_for_each_gt(pf_fd, gt) {
-		for (vf = 0; vf <= num_vfs; ++vf) {
-			ret = __xe_sriov_set_exec_quantum_ms(pf_fd, vf, gt,
-							     p->exec_quantum_ms);
-			if (igt_warn_on_f(ret,
-					  "Failed to set exec_quantum_ms=%u (VF%d GT%u): %d\n",
-					  p->exec_quantum_ms, vf, gt, ret))
-				return ret;
-
-			ret = __xe_sriov_set_preempt_timeout_us(pf_fd, vf, gt,
-								p->preempt_timeout_us);
-			if (igt_warn_on_f(ret,
-					  "Failed to set preempt_timeout_us=%u (VF%d GT%u): %d\n",
-					  p->preempt_timeout_us, vf, gt, ret))
-				return ret;
+	int ret = 0;
+
+	if (xe_sriov_admin_is_present(pf_fd)) {
+		ret = __xe_sriov_bulk_set_exec_quantum_ms(pf_fd, p->exec_quantum_ms);
+		if (igt_warn_on_f(ret,
+				  "Failed to bulk set exec quantum=%u: %d\n",
+				  p->exec_quantum_ms, ret))
+			return ret;
+
+		ret = __xe_sriov_bulk_set_preempt_timeout_us(pf_fd, p->preempt_timeout_us);
+		if (igt_warn_on_f(ret,
+				  "Failed to bulk set preempt timeout=%u: %d\n",
+				  p->preempt_timeout_us, ret))
+			return ret;
+	} else {
+		unsigned int gt;
+
+		xe_for_each_gt(pf_fd, gt) {
+			for (int vf = 0; vf <= num_vfs; ++vf) {
+				ret = __xe_sriov_set_exec_quantum_ms(pf_fd, vf, gt,
+								     p->exec_quantum_ms);
+				if (igt_warn_on_f(ret,
+						  "Failed to set exec quantum=%u (VF%d GT%u): %d\n",
+						  p->exec_quantum_ms, vf, gt, ret))
+					return ret;
+
+				ret = __xe_sriov_set_preempt_timeout_us(pf_fd, vf, gt,
+									p->preempt_timeout_us);
+				if (igt_warn_on_f(ret,
+						  "Failed to set preempt timeout=%u (VF%d GT%u): %d\n",
+						  p->preempt_timeout_us, vf, gt, ret))
+					return ret;
+			}
 		}
 	}
-	return 0;
+
+	return ret;
 }
 
 static void set_vfs_scheduling_params(int pf_fd, int num_vfs,
-- 
2.43.0