#include "xe_wa.h"
 
+#include <drm/drm_managed.h>
 #include <kunit/visibility.h>
 #include <linux/compiler_types.h>
 
 {
        struct xe_rtp_process_ctx ctx = XE_RTP_PROCESS_CTX_INITIALIZER(gt);
 
+       xe_rtp_process_ctx_enable_active_tracking(&ctx, gt->wa_active.gt,
+                                                 ARRAY_SIZE(gt_was));
        xe_rtp_process_to_sr(&ctx, gt_was, >->reg_sr);
 }
 EXPORT_SYMBOL_IF_KUNIT(xe_wa_process_gt);
 {
        struct xe_rtp_process_ctx ctx = XE_RTP_PROCESS_CTX_INITIALIZER(hwe);
 
+       xe_rtp_process_ctx_enable_active_tracking(&ctx, hwe->gt->wa_active.engine,
+                                                 ARRAY_SIZE(engine_was));
        xe_rtp_process_to_sr(&ctx, engine_was, &hwe->reg_sr);
 }
 
 {
        struct xe_rtp_process_ctx ctx = XE_RTP_PROCESS_CTX_INITIALIZER(hwe);
 
+       xe_rtp_process_ctx_enable_active_tracking(&ctx, hwe->gt->wa_active.lrc,
+                                                 ARRAY_SIZE(lrc_was));
        xe_rtp_process_to_sr(&ctx, lrc_was, &hwe->reg_lrc);
 }
+
+/**
+ * xe_wa_init - initialize gt with workaround bookkeeping
+ * @gt: GT instance to initialize
+ *
+ * Returns 0 for success, negative error code otherwise.
+ */
+int xe_wa_init(struct xe_gt *gt)
+{
+       struct xe_device *xe = gt_to_xe(gt);
+       size_t n_lrc, n_engine, n_gt, total;
+       unsigned long *p;
+
+       n_gt = BITS_TO_LONGS(ARRAY_SIZE(gt_was));
+       n_engine = BITS_TO_LONGS(ARRAY_SIZE(engine_was));
+       n_lrc = BITS_TO_LONGS(ARRAY_SIZE(lrc_was));
+       total = n_gt + n_engine + n_lrc;
+
+       p = drmm_kzalloc(&xe->drm, sizeof(*p) * total, GFP_KERNEL);
+       if (!p)
+               return -ENOMEM;
+
+       gt->wa_active.gt = p;
+       p += n_gt;
+       gt->wa_active.engine = p;
+       p += n_engine;
+       gt->wa_active.lrc = p;
+
+       return 0;
+}