#include "gem/i915_gem_context.h"
 #include "gem/i915_gem_pm.h"
+#include "gt/intel_engine_user.h"
 #include "gt/intel_lrc_reg.h"
 
 #include "i915_drv.h"
  * @oa_format: An OA unit HW report format
  * @oa_periodic: Whether to enable periodic OA unit sampling
  * @oa_period_exponent: The OA unit sampling period is derived from this
+ * @engine: The engine (typically rcs0) being monitored by the OA unit
  *
  * As read_properties_unlocked() enumerates and validates the properties given
  * to open a stream of metrics the configuration is built up in the structure
        int oa_format;
        bool oa_periodic;
        int oa_period_exponent;
+
+       struct intel_engine_cs *engine;
 };
 
 static enum hrtimer_restart oa_poll_check_timer_cb(struct hrtimer *hrtimer);
        int err;
 
        for_each_gem_engine(ce, i915_gem_context_lock_engines(ctx), it) {
-               if (ce->engine->class != RENDER_CLASS)
+               if (ce->engine != stream->engine) /* first match! */
                        continue;
 
                /*
        int format_size;
        int ret;
 
-       /* If the sysfs metrics/ directory wasn't registered for some
+       if (!props->engine) {
+               DRM_DEBUG("OA engine not specified\n");
+               return -EINVAL;
+       }
+
+       /*
+        * If the sysfs metrics/ directory wasn't registered for some
         * reason then don't let userspace try their luck with config
         * IDs
         */
                return -ENODEV;
        }
 
-       /* To avoid the complexity of having to accurately filter
+       /*
+        * To avoid the complexity of having to accurately filter
         * counter reports and marshal to the appropriate client
         * we currently only allow exclusive access
         */
                return -EINVAL;
        }
 
+       stream->engine = props->engine;
+       stream->gt = stream->engine->gt;
+
        stream->sample_size = sizeof(struct drm_i915_perf_record_header);
 
        format_size = perf->oa_formats[props->oa_format].size;
        }
 
        stream->perf = perf;
-       stream->gt = &perf->i915->gt;
        stream->ctx = specific_ctx;
 
        ret = i915_oa_stream_init(stream, param, props);
                return -EINVAL;
        }
 
+       /* At the moment we only support using i915-perf on the RCS. */
+       props->engine = intel_engine_lookup_user(perf->i915,
+                                                I915_ENGINE_CLASS_RENDER,
+                                                0);
+       if (!props->engine) {
+               DRM_DEBUG("No RENDER-capable engines\n");
+               return -EINVAL;
+       }
+
        /* Considering that ID = 0 is reserved and assuming that we don't
         * (currently) expect any configurations to ever specify duplicate
         * values for a particular property ID then the last _PROP_MAX value is