We want to pass in a intel_context into intel_context_pin() and that
requires us to first be able to lookup the intel_context!

Signed-off-by: Chris Wilson <[email protected]>
---
 drivers/gpu/drm/i915/gvt/scheduler.c       |  7 +++-
 drivers/gpu/drm/i915/i915_gem_execbuffer.c | 11 +++++--
 drivers/gpu/drm/i915/i915_perf.c           | 21 ++++++++----
 drivers/gpu/drm/i915/i915_request.c        |  8 ++++-
 drivers/gpu/drm/i915/intel_context.c       | 38 +++++++++++-----------
 drivers/gpu/drm/i915/intel_context.h       | 19 +++++++----
 drivers/gpu/drm/i915/intel_engine_cs.c     |  8 ++++-
 7 files changed, 74 insertions(+), 38 deletions(-)

diff --git a/drivers/gpu/drm/i915/gvt/scheduler.c 
b/drivers/gpu/drm/i915/gvt/scheduler.c
index 2d2dafd22a18..ae1f09d2d4ae 100644
--- a/drivers/gpu/drm/i915/gvt/scheduler.c
+++ b/drivers/gpu/drm/i915/gvt/scheduler.c
@@ -1191,12 +1191,17 @@ int intel_vgpu_setup_submission(struct intel_vgpu *vgpu)
                INIT_LIST_HEAD(&s->workload_q_head[i]);
                s->shadow[i] = ERR_PTR(-EINVAL);
 
-               ce = intel_context_pin(ctx, engine);
+               ce = intel_context_instance(ctx, engine);
                if (IS_ERR(ce)) {
                        ret = PTR_ERR(ce);
                        goto out_shadow_ctx;
                }
 
+               ret = intel_context_pin(ce);
+               intel_context_put(ce);
+               if (ret)
+                       goto out_shadow_ctx;
+
                s->shadow[i] = ce;
        }
 
diff --git a/drivers/gpu/drm/i915/i915_gem_execbuffer.c 
b/drivers/gpu/drm/i915/i915_gem_execbuffer.c
index 8754bb02c6ec..cf7aa0e325d2 100644
--- a/drivers/gpu/drm/i915/i915_gem_execbuffer.c
+++ b/drivers/gpu/drm/i915/i915_gem_execbuffer.c
@@ -2099,14 +2099,19 @@ static int eb_pin_context(struct i915_execbuffer *eb,
        if (err)
                return err;
 
+       ce = intel_context_instance(eb->gem_context, engine);
+       if (IS_ERR(ce))
+               return PTR_ERR(ce);
+
        /*
         * Pinning the contexts may generate requests in order to acquire
         * GGTT space, so do this first before we reserve a seqno for
         * ourselves.
         */
-       ce = intel_context_pin(eb->gem_context, engine);
-       if (IS_ERR(ce))
-               return PTR_ERR(ce);
+       err = intel_context_pin(ce);
+       intel_context_put(ce);
+       if (err)
+               return err;
 
        eb->engine = engine;
        eb->context = ce;
diff --git a/drivers/gpu/drm/i915/i915_perf.c b/drivers/gpu/drm/i915/i915_perf.c
index fe7267da52e5..28475cbbdcbb 100644
--- a/drivers/gpu/drm/i915/i915_perf.c
+++ b/drivers/gpu/drm/i915/i915_perf.c
@@ -1205,11 +1205,17 @@ static struct intel_context *oa_pin_context(struct 
drm_i915_private *i915,
 {
        struct intel_engine_cs *engine = i915->engine[RCS0];
        struct intel_context *ce;
-       int ret;
+       int err;
 
-       ret = i915_mutex_lock_interruptible(&i915->drm);
-       if (ret)
-               return ERR_PTR(ret);
+       ce = intel_context_instance(ctx, engine);
+       if (IS_ERR(ce))
+               return ce;
+
+       err = i915_mutex_lock_interruptible(&i915->drm);
+       if (err) {
+               intel_context_put(ce);
+               return ERR_PTR(err);
+       }
 
        /*
         * As the ID is the gtt offset of the context's vma we
@@ -1217,10 +1223,11 @@ static struct intel_context *oa_pin_context(struct 
drm_i915_private *i915,
         *
         * NB: implied RCS engine...
         */
-       ce = intel_context_pin(ctx, engine);
+       err = intel_context_pin(ce);
        mutex_unlock(&i915->drm.struct_mutex);
-       if (IS_ERR(ce))
-               return ce;
+       intel_context_put(ce);
+       if (err)
+               return ERR_PTR(err);
 
        i915->perf.oa.pinned_ctx = ce;
 
diff --git a/drivers/gpu/drm/i915/i915_request.c 
b/drivers/gpu/drm/i915/i915_request.c
index 10edeb285870..fe8db5ef0ded 100644
--- a/drivers/gpu/drm/i915/i915_request.c
+++ b/drivers/gpu/drm/i915/i915_request.c
@@ -741,6 +741,7 @@ i915_request_alloc(struct intel_engine_cs *engine, struct 
i915_gem_context *ctx)
        struct drm_i915_private *i915 = engine->i915;
        struct intel_context *ce;
        struct i915_request *rq;
+       int err;
 
        /*
         * Preempt contexts are reserved for exclusive use to inject a
@@ -754,10 +755,15 @@ i915_request_alloc(struct intel_engine_cs *engine, struct 
i915_gem_context *ctx)
         * GGTT space, so do this first before we reserve a seqno for
         * ourselves.
         */
-       ce = intel_context_pin(ctx, engine);
+       ce = intel_context_instance(ctx, engine);
        if (IS_ERR(ce))
                return ERR_CAST(ce);
 
+       err = intel_context_pin(ce);
+       intel_context_put(ce);
+       if (err)
+               return ERR_PTR(err);
+
        i915_gem_unpark(i915);
 
        rq = i915_request_create(ce);
diff --git a/drivers/gpu/drm/i915/intel_context.c 
b/drivers/gpu/drm/i915/intel_context.c
index 8931e0fee873..ca81b4dc5364 100644
--- a/drivers/gpu/drm/i915/intel_context.c
+++ b/drivers/gpu/drm/i915/intel_context.c
@@ -102,7 +102,7 @@ void __intel_context_remove(struct intel_context *ce)
        spin_unlock(&ctx->hw_contexts_lock);
 }
 
-static struct intel_context *
+struct intel_context *
 intel_context_instance(struct i915_gem_context *ctx,
                       struct intel_engine_cs *engine)
 {
@@ -110,7 +110,7 @@ intel_context_instance(struct i915_gem_context *ctx,
 
        ce = intel_context_lookup(ctx, engine);
        if (likely(ce))
-               return ce;
+               return intel_context_get(ce);
 
        ce = intel_context_alloc();
        if (!ce)
@@ -123,7 +123,7 @@ intel_context_instance(struct i915_gem_context *ctx,
                intel_context_free(ce);
 
        GEM_BUG_ON(intel_context_lookup(ctx, engine) != pos);
-       return pos;
+       return intel_context_get(pos);
 }
 
 struct intel_context *
@@ -137,36 +137,36 @@ intel_context_pin_lock(struct i915_gem_context *ctx,
        if (IS_ERR(ce))
                return ce;
 
-       if (mutex_lock_interruptible(&ce->pin_mutex))
+       if (mutex_lock_interruptible(&ce->pin_mutex)) {
+               intel_context_put(ce);
                return ERR_PTR(-EINTR);
+       }
 
        return ce;
 }
 
-struct intel_context *
-intel_context_pin(struct i915_gem_context *ctx,
-                 struct intel_engine_cs *engine)
+void intel_context_pin_unlock(struct intel_context *ce)
+       __releases(ce->pin_mutex)
 {
-       struct intel_context *ce;
-       int err;
-
-       ce = intel_context_instance(ctx, engine);
-       if (IS_ERR(ce))
-               return ce;
+       mutex_unlock(&ce->pin_mutex);
+       intel_context_put(ce);
+}
 
-       if (likely(atomic_inc_not_zero(&ce->pin_count)))
-               return ce;
+int __intel_context_do_pin(struct intel_context *ce)
+{
+       int err;
 
        if (mutex_lock_interruptible(&ce->pin_mutex))
-               return ERR_PTR(-EINTR);
+               return -EINTR;
 
        if (likely(!atomic_read(&ce->pin_count))) {
+               struct i915_gem_context *ctx = ce->gem_context;
+
                err = ce->ops->pin(ce);
                if (err)
                        goto err;
 
                i915_gem_context_get(ctx);
-               GEM_BUG_ON(ce->gem_context != ctx);
 
                mutex_lock(&ctx->mutex);
                list_add(&ce->active_link, &ctx->active_engines);
@@ -180,11 +180,11 @@ intel_context_pin(struct i915_gem_context *ctx,
        GEM_BUG_ON(!intel_context_is_pinned(ce)); /* no overflow! */
 
        mutex_unlock(&ce->pin_mutex);
-       return ce;
+       return 0;
 
 err:
        mutex_unlock(&ce->pin_mutex);
-       return ERR_PTR(err);
+       return err;
 }
 
 void intel_context_unpin(struct intel_context *ce)
diff --git a/drivers/gpu/drm/i915/intel_context.h 
b/drivers/gpu/drm/i915/intel_context.h
index ebc861b1a49e..2daf6a5217ae 100644
--- a/drivers/gpu/drm/i915/intel_context.h
+++ b/drivers/gpu/drm/i915/intel_context.h
@@ -49,11 +49,7 @@ intel_context_is_pinned(struct intel_context *ce)
        return atomic_read(&ce->pin_count);
 }
 
-static inline void intel_context_pin_unlock(struct intel_context *ce)
-__releases(ce->pin_mutex)
-{
-       mutex_unlock(&ce->pin_mutex);
-}
+void intel_context_pin_unlock(struct intel_context *ce);
 
 struct intel_context *
 __intel_context_insert(struct i915_gem_context *ctx,
@@ -63,7 +59,18 @@ void
 __intel_context_remove(struct intel_context *ce);
 
 struct intel_context *
-intel_context_pin(struct i915_gem_context *ctx, struct intel_engine_cs 
*engine);
+intel_context_instance(struct i915_gem_context *ctx,
+                      struct intel_engine_cs *engine);
+
+int __intel_context_do_pin(struct intel_context *ce);
+
+static inline int intel_context_pin(struct intel_context *ce)
+{
+       if (likely(atomic_inc_not_zero(&ce->pin_count)))
+               return 0;
+
+       return __intel_context_do_pin(ce);
+}
 
 static inline void __intel_context_pin(struct intel_context *ce)
 {
diff --git a/drivers/gpu/drm/i915/intel_engine_cs.c 
b/drivers/gpu/drm/i915/intel_engine_cs.c
index c5b417327132..a7413e74410f 100644
--- a/drivers/gpu/drm/i915/intel_engine_cs.c
+++ b/drivers/gpu/drm/i915/intel_engine_cs.c
@@ -693,11 +693,17 @@ static int pin_context(struct i915_gem_context *ctx,
                       struct intel_context **out)
 {
        struct intel_context *ce;
+       int err;
 
-       ce = intel_context_pin(ctx, engine);
+       ce = intel_context_instance(ctx, engine);
        if (IS_ERR(ce))
                return PTR_ERR(ce);
 
+       err = intel_context_pin(ce);
+       intel_context_put(ce);
+       if (err)
+               return err;
+
        *out = ce;
        return 0;
 }
-- 
2.20.1

_______________________________________________
Intel-gfx mailing list
[email protected]
https://lists.freedesktop.org/mailman/listinfo/intel-gfx

Reply via email to