2 * SPDX-License-Identifier: MIT
4 * Copyright © 2019 Intel Corporation
7 #include "gem/i915_gem_context.h"
8 #include "gem/i915_gem_pm.h"
11 #include "i915_globals.h"
13 #include "intel_context.h"
14 #include "intel_engine.h"
15 #include "intel_engine_pm.h"
17 static struct i915_global_context {
18 struct i915_global base;
19 struct kmem_cache *slab_ce;
22 static struct intel_context *intel_context_alloc(void)
24 return kmem_cache_zalloc(global.slab_ce, GFP_KERNEL);
27 void intel_context_free(struct intel_context *ce)
29 kmem_cache_free(global.slab_ce, ce);
32 struct intel_context *
33 intel_context_create(struct i915_gem_context *ctx,
34 struct intel_engine_cs *engine)
36 struct intel_context *ce;
38 ce = intel_context_alloc();
40 return ERR_PTR(-ENOMEM);
42 intel_context_init(ce, ctx, engine);
46 int __intel_context_do_pin(struct intel_context *ce)
50 if (mutex_lock_interruptible(&ce->pin_mutex))
53 if (likely(!atomic_read(&ce->pin_count))) {
54 intel_wakeref_t wakeref;
57 with_intel_runtime_pm(&ce->engine->i915->runtime_pm, wakeref)
58 err = ce->ops->pin(ce);
62 i915_gem_context_get(ce->gem_context); /* for ctx->ppgtt */
64 smp_mb__before_atomic(); /* flush pin before it is visible */
67 atomic_inc(&ce->pin_count);
68 GEM_BUG_ON(!intel_context_is_pinned(ce)); /* no overflow! */
70 mutex_unlock(&ce->pin_mutex);
74 mutex_unlock(&ce->pin_mutex);
78 void intel_context_unpin(struct intel_context *ce)
80 if (likely(atomic_add_unless(&ce->pin_count, -1, 1)))
83 /* We may be called from inside intel_context_pin() to evict another */
84 intel_context_get(ce);
85 mutex_lock_nested(&ce->pin_mutex, SINGLE_DEPTH_NESTING);
87 if (likely(atomic_dec_and_test(&ce->pin_count))) {
90 i915_gem_context_put(ce->gem_context);
91 intel_context_active_release(ce);
94 mutex_unlock(&ce->pin_mutex);
95 intel_context_put(ce);
98 static int __context_pin_state(struct i915_vma *vma, unsigned long flags)
102 err = i915_vma_pin(vma, 0, 0, flags | PIN_GLOBAL);
107 * And mark it as a globally pinned object to let the shrinker know
108 * it cannot reclaim the object until we release it.
110 vma->obj->pin_global++;
111 vma->obj->mm.dirty = true;
116 static void __context_unpin_state(struct i915_vma *vma)
118 vma->obj->pin_global--;
119 __i915_vma_unpin(vma);
122 static void intel_context_retire(struct i915_active *active)
124 struct intel_context *ce = container_of(active, typeof(*ce), active);
127 __context_unpin_state(ce->state);
129 intel_ring_unpin(ce->ring);
130 intel_context_put(ce);
134 intel_context_init(struct intel_context *ce,
135 struct i915_gem_context *ctx,
136 struct intel_engine_cs *engine)
138 GEM_BUG_ON(!engine->cops);
142 ce->gem_context = ctx;
144 ce->ops = engine->cops;
145 ce->sseu = engine->sseu;
147 INIT_LIST_HEAD(&ce->signal_link);
148 INIT_LIST_HEAD(&ce->signals);
150 mutex_init(&ce->pin_mutex);
152 i915_active_init(ctx->i915, &ce->active, intel_context_retire);
155 int intel_context_active_acquire(struct intel_context *ce, unsigned long flags)
159 if (!i915_active_acquire(&ce->active))
162 intel_context_get(ce);
164 err = intel_ring_pin(ce->ring);
171 err = __context_pin_state(ce->state, flags);
175 /* Preallocate tracking nodes */
176 if (!i915_gem_context_is_kernel(ce->gem_context)) {
177 err = i915_active_acquire_preallocate_barrier(&ce->active,
186 __context_unpin_state(ce->state);
188 intel_ring_unpin(ce->ring);
190 intel_context_put(ce);
191 i915_active_cancel(&ce->active);
195 void intel_context_active_release(struct intel_context *ce)
197 /* Nodes preallocated in intel_context_active() */
198 i915_active_acquire_barrier(&ce->active);
199 i915_active_release(&ce->active);
202 static void i915_global_context_shrink(void)
204 kmem_cache_shrink(global.slab_ce);
207 static void i915_global_context_exit(void)
209 kmem_cache_destroy(global.slab_ce);
212 static struct i915_global_context global = { {
213 .shrink = i915_global_context_shrink,
214 .exit = i915_global_context_exit,
217 int __init i915_global_context_init(void)
219 global.slab_ce = KMEM_CACHE(intel_context, SLAB_HWCACHE_ALIGN);
223 i915_global_register(&global.base);
227 void intel_context_enter_engine(struct intel_context *ce)
229 intel_engine_pm_get(ce->engine);
232 void intel_context_exit_engine(struct intel_context *ce)
234 intel_engine_pm_put(ce->engine);
237 struct i915_request *intel_context_create_request(struct intel_context *ce)
239 struct i915_request *rq;
242 err = intel_context_pin(ce);
246 rq = i915_request_create(ce);
247 intel_context_unpin(ce);