[PATCH i-g-t 1/2] lib/xe/xe_query: Don’t assume sequential GT IDs in xe_gt_type()

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

Replace direct gt_list indexing with drm_xe_get_gt() and check
gt_mask before use. This ensures xe_gt_type() works with real
GT IDs instead of assuming list index == uapi ID, consistent
with other helpers.

Fixes: 3bb0c49ac668 ("lib/xe/xe_query: Add xe_gt_type() and xe_is_main_gt() helpers")

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: Kamil Konieczny <kamil.konieczny@linux.intel.com>
Cc: Lukasz Laguna <lukasz.laguna@intel.com>
Cc: Piotr Piórkowski <piotr.piorkowski@intel.com>
Cc: Satyanarayana K V P <satyanarayana.k.v.p@intel.com>
---
 lib/xe/xe_query.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/lib/xe/xe_query.c b/lib/xe/xe_query.c
index 9b88cce8d..17045f3fb 100644
--- a/lib/xe/xe_query.c
+++ b/lib/xe/xe_query.c
@@ -927,12 +927,15 @@ bool xe_has_media_gt(int fd)
 uint16_t xe_gt_type(int fd, int gt)
 {
 	struct xe_device *xe_dev = find_in_cache(fd);
+	const struct drm_xe_gt *xe_gt;
 
 	igt_assert(xe_dev);
-	igt_assert_f(gt >= 0 && gt < xe_number_gt(fd),
-		     "gt %d out of range [0..%d)\n", gt, xe_number_gt(fd));
+	igt_assert(xe_dev->gt_mask & BIT(gt));
+
+	xe_gt = drm_xe_get_gt(xe_dev, gt);
+	igt_assert(xe_gt);
 
-	return xe_dev->gt_list->gt_list[gt].type;
+	return xe_gt->type;
 }
 
 /**
-- 
2.31.1