struct intel_engine_cs *engine = NULL;
        struct i915_request *rq;
 
+       if (intel_context_has_inflight(ce))
+               return intel_context_inflight(ce);
+
        if (!ce->timeline)
                return NULL;
 
 
        return test_bit(CONTEXT_CLOSED_BIT, &ce->flags);
 }
 
+static inline bool intel_context_has_inflight(const struct intel_context *ce)
+{
+       return test_bit(COPS_HAS_INFLIGHT_BIT, &ce->ops->flags);
+}
+
 static inline bool intel_context_use_semaphores(const struct intel_context *ce)
 {
        return test_bit(CONTEXT_USE_SEMAPHORES, &ce->flags);
 
 struct intel_ring;
 
 struct intel_context_ops {
+       unsigned long flags;
+#define COPS_HAS_INFLIGHT_BIT 0
+#define COPS_HAS_INFLIGHT BIT(COPS_HAS_INFLIGHT_BIT)
+
        int (*alloc)(struct intel_context *ce);
 
        int (*pre_pin)(struct intel_context *ce, struct i915_gem_ww_ctx *ww, void **vaddr);
 
 }
 
 static const struct intel_context_ops execlists_context_ops = {
+       .flags = COPS_HAS_INFLIGHT,
+
        .alloc = execlists_context_alloc,
 
        .pre_pin = execlists_context_pre_pin,
 }
 
 static const struct intel_context_ops virtual_context_ops = {
+       .flags = COPS_HAS_INFLIGHT,
+
        .alloc = virtual_context_alloc,
 
        .pre_pin = virtual_context_pre_pin,