return __context_set_persistence(ctx, args->value);
 }
 
+static void __apply_priority(struct intel_context *ce, void *arg)
+{
+       struct i915_gem_context *ctx = arg;
+
+       if (!intel_engine_has_semaphores(ce->engine))
+               return;
+
+       if (ctx->sched.priority >= I915_PRIORITY_NORMAL)
+               intel_context_set_use_semaphores(ce);
+       else
+               intel_context_clear_use_semaphores(ce);
+}
+
+static int set_priority(struct i915_gem_context *ctx,
+                       const struct drm_i915_gem_context_param *args)
+{
+       s64 priority = args->value;
+
+       if (args->size)
+               return -EINVAL;
+
+       if (!(ctx->i915->caps.scheduler & I915_SCHEDULER_CAP_PRIORITY))
+               return -ENODEV;
+
+       if (priority > I915_CONTEXT_MAX_USER_PRIORITY ||
+           priority < I915_CONTEXT_MIN_USER_PRIORITY)
+               return -EINVAL;
+
+       if (priority > I915_CONTEXT_DEFAULT_PRIORITY &&
+           !capable(CAP_SYS_NICE))
+               return -EPERM;
+
+       ctx->sched.priority = I915_USER_PRIORITY(priority);
+       context_apply_all(ctx, __apply_priority, ctx);
+
+       return 0;
+}
+
 static int ctx_setparam(struct drm_i915_file_private *fpriv,
                        struct i915_gem_context *ctx,
                        struct drm_i915_gem_context_param *args)
                break;
 
        case I915_CONTEXT_PARAM_PRIORITY:
-               {
-                       s64 priority = args->value;
-
-                       if (args->size)
-                               ret = -EINVAL;
-                       else if (!(ctx->i915->caps.scheduler & I915_SCHEDULER_CAP_PRIORITY))
-                               ret = -ENODEV;
-                       else if (priority > I915_CONTEXT_MAX_USER_PRIORITY ||
-                                priority < I915_CONTEXT_MIN_USER_PRIORITY)
-                               ret = -EINVAL;
-                       else if (priority > I915_CONTEXT_DEFAULT_PRIORITY &&
-                                !capable(CAP_SYS_NICE))
-                               ret = -EPERM;
-                       else
-                               ctx->sched.priority =
-                                       I915_USER_PRIORITY(priority);
-               }
+               ret = set_priority(ctx, args);
                break;
 
        case I915_CONTEXT_PARAM_SSEU:
 
        return u64_to_ptr(struct intel_ring, sz);
 }
 
+static inline bool intel_context_use_semaphores(const struct intel_context *ce)
+{
+       return test_bit(CONTEXT_USE_SEMAPHORES, &ce->flags);
+}
+
+static inline void intel_context_set_use_semaphores(struct intel_context *ce)
+{
+       set_bit(CONTEXT_USE_SEMAPHORES, &ce->flags);
+}
+
+static inline void intel_context_clear_use_semaphores(struct intel_context *ce)
+{
+       clear_bit(CONTEXT_USE_SEMAPHORES, &ce->flags);
+}
+
 static inline bool intel_context_is_banned(const struct intel_context *ce)
 {
        return test_bit(CONTEXT_BANNED, &ce->flags);
 
        unsigned long flags;
 #define CONTEXT_ALLOC_BIT              0
 #define CONTEXT_VALID_BIT              1
-#define CONTEXT_BANNED                 2
-#define CONTEXT_FORCE_SINGLE_SUBMISSION        3
-#define CONTEXT_NOPREEMPT              4
+#define CONTEXT_USE_SEMAPHORES         2
+#define CONTEXT_BANNED                 3
+#define CONTEXT_FORCE_SINGLE_SUBMISSION        4
+#define CONTEXT_NOPREEMPT              5
 
        u32 *lrc_reg_state;
        u64 lrc_desc;
 
                        return ret;
        }
 
-       if (to->engine == from->engine) {
+       if (to->engine == from->engine)
                ret = i915_sw_fence_await_sw_fence_gfp(&to->submit,
                                                       &from->submit,
                                                       I915_FENCE_GFP);
-       } else if (intel_engine_has_semaphores(to->engine) &&
-                  to->context->gem_context->sched.priority >= I915_PRIORITY_NORMAL) {
+       else if (intel_context_use_semaphores(to->context))
                ret = emit_semaphore_wait(to, from, I915_FENCE_GFP);
-       } else {
+       else
                ret = i915_sw_fence_await_dma_fence(&to->submit,
                                                    &from->fence, 0,
                                                    I915_FENCE_GFP);
-       }
        if (ret < 0)
                return ret;