[PATCH v3 3/7] drm/xe/xe_device: Reintroduce xe_device_get_gt

[Jonathan Cavitt <jonathan.cavitt@intel.com>]

Reintroduce the function xe_device_get_gt, as it was earlier renamed to
xe_device_lookup_gt.  It performs the same functionality as
xe_device_lookup_gt, but it guarantees the xe_gt pointer returned is not
NULL using an assertion.

Suggested-by: Rodrigo Vivi <rodrigo.vivi@intel.com>
Suggested-by: Michal Wajdeczko <michal.wajdeczko@intel.com>
Signed-off-by: Jonathan Cavitt <jonathan.cavitt@intel.com>
---
 drivers/gpu/drm/xe/xe_device.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/drivers/gpu/drm/xe/xe_device.h b/drivers/gpu/drm/xe/xe_device.h
index 807db49a5306..af9476879454 100644
--- a/drivers/gpu/drm/xe/xe_device.h
+++ b/drivers/gpu/drm/xe/xe_device.h
@@ -97,6 +97,21 @@ static inline struct xe_gt *xe_device_lookup_gt(struct xe_device *xe, u8 gt_id)
 	return gt;
 }
 
+/**
+ * xe_device_get_gt() - Same as xe_device_lookup_gt, but return must not be NULL
+ * @xe: The XE device to search
+ * @gt_id: The ID of the GT to search for
+ *
+ * Returns a pointer to the xe_gt struct with the given gt_id.
+ */
+static inline struct xe_gt *xe_device_get_gt(struct xe_device *xe, u8 gt_id)
+{
+	struct xe_gt *gt = xe_device_lookup_gt(xe, gt_id);
+
+	xe_assert(xe, gt);
+	return gt;
+}
+
 /*
  * Provide a GT structure suitable for performing non-GT MMIO operations against
  * the primary tile.  Primarily intended for early tile initialization, display
-- 
2.43.0