When cloning the engines from the source context, we need to ensure that
the engines are not freed as we copy them, and that the flags we clone
from the source correspond with the engines we copy across. To do this
we need only take a reference to the src->engines, rather than hold the
src->engine_mutex, so long as we verify that nothing changed under the
read.

Signed-off-by: Chris Wilson <ch...@chris-wilson.co.uk>
---
 drivers/gpu/drm/i915/gem/i915_gem_context.c | 24 +++++++++++++--------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/drivers/gpu/drm/i915/gem/i915_gem_context.c 
b/drivers/gpu/drm/i915/gem/i915_gem_context.c
index db893f6c516b..6530bd2e634e 100644
--- a/drivers/gpu/drm/i915/gem/i915_gem_context.c
+++ b/drivers/gpu/drm/i915/gem/i915_gem_context.c
@@ -755,7 +755,8 @@ __create_context(struct drm_i915_private *i915)
 }
 
 static inline struct i915_gem_engines *
-__context_engines_await(const struct i915_gem_context *ctx)
+__context_engines_await(const struct i915_gem_context *ctx,
+                       bool *user_engines)
 {
        struct i915_gem_engines *engines;
 
@@ -764,6 +765,10 @@ __context_engines_await(const struct i915_gem_context *ctx)
                engines = rcu_dereference(ctx->engines);
                GEM_BUG_ON(!engines);
 
+               if (user_engines)
+                       *user_engines = i915_gem_context_user_engines(ctx);
+
+               /* successful await => strong mb */
                if (unlikely(!i915_sw_fence_await(&engines->fence)))
                        continue;
 
@@ -787,7 +792,7 @@ context_apply_all(struct i915_gem_context *ctx,
        struct intel_context *ce;
        int err = 0;
 
-       e = __context_engines_await(ctx);
+       e = __context_engines_await(ctx, NULL);
        for_each_gem_engine(ce, e, it) {
                err = fn(ce, data);
                if (err)
@@ -1129,7 +1134,7 @@ static int context_barrier_task(struct i915_gem_context 
*ctx,
                return err;
        }
 
-       e = __context_engines_await(ctx);
+       e = __context_engines_await(ctx, NULL);
        if (!e) {
                i915_active_release(&cb->base);
                return -ENOENT;
@@ -2126,11 +2131,14 @@ static int copy_ring_size(struct intel_context *dst,
 static int clone_engines(struct i915_gem_context *dst,
                         struct i915_gem_context *src)
 {
-       struct i915_gem_engines *e = i915_gem_context_lock_engines(src);
-       struct i915_gem_engines *clone;
+       struct i915_gem_engines *clone, *e;
        bool user_engines;
        unsigned long n;
 
+       e = __context_engines_await(src, &user_engines);
+       if (!e)
+               return -ENOENT;
+
        clone = alloc_engines(e->num_engines);
        if (!clone)
                goto err_unlock;
@@ -2172,9 +2180,7 @@ static int clone_engines(struct i915_gem_context *dst,
                }
        }
        clone->num_engines = n;
-
-       user_engines = i915_gem_context_user_engines(src);
-       i915_gem_context_unlock_engines(src);
+       i915_sw_fence_complete(&e->fence);
 
        /* Serialised by constructor */
        engines_idle_release(dst, rcu_replace_pointer(dst->engines, clone, 1));
@@ -2185,7 +2191,7 @@ static int clone_engines(struct i915_gem_context *dst,
        return 0;
 
 err_unlock:
-       i915_gem_context_unlock_engines(src);
+       i915_sw_fence_complete(&e->fence);
        return -ENOMEM;
 }
 
-- 
2.20.1

_______________________________________________
Intel-gfx mailing list
Intel-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/intel-gfx

Reply via email to