110be98a7SChris Wilson /* 210be98a7SChris Wilson * SPDX-License-Identifier: MIT 310be98a7SChris Wilson * 410be98a7SChris Wilson * Copyright © 2011-2012 Intel Corporation 510be98a7SChris Wilson */ 610be98a7SChris Wilson 710be98a7SChris Wilson /* 810be98a7SChris Wilson * This file implements HW context support. On gen5+ a HW context consists of an 910be98a7SChris Wilson * opaque GPU object which is referenced at times of context saves and restores. 1010be98a7SChris Wilson * With RC6 enabled, the context is also referenced as the GPU enters and exists 1110be98a7SChris Wilson * from RC6 (GPU has it's own internal power context, except on gen5). Though 1210be98a7SChris Wilson * something like a context does exist for the media ring, the code only 1310be98a7SChris Wilson * supports contexts for the render ring. 1410be98a7SChris Wilson * 1510be98a7SChris Wilson * In software, there is a distinction between contexts created by the user, 1610be98a7SChris Wilson * and the default HW context. The default HW context is used by GPU clients 1710be98a7SChris Wilson * that do not request setup of their own hardware context. The default 1810be98a7SChris Wilson * context's state is never restored to help prevent programming errors. This 1910be98a7SChris Wilson * would happen if a client ran and piggy-backed off another clients GPU state. 2010be98a7SChris Wilson * The default context only exists to give the GPU some offset to load as the 2110be98a7SChris Wilson * current to invoke a save of the context we actually care about. In fact, the 2210be98a7SChris Wilson * code could likely be constructed, albeit in a more complicated fashion, to 2310be98a7SChris Wilson * never use the default context, though that limits the driver's ability to 2410be98a7SChris Wilson * swap out, and/or destroy other contexts. 2510be98a7SChris Wilson * 2610be98a7SChris Wilson * All other contexts are created as a request by the GPU client. These contexts 2710be98a7SChris Wilson * store GPU state, and thus allow GPU clients to not re-emit state (and 2810be98a7SChris Wilson * potentially query certain state) at any time. The kernel driver makes 2910be98a7SChris Wilson * certain that the appropriate commands are inserted. 3010be98a7SChris Wilson * 3110be98a7SChris Wilson * The context life cycle is semi-complicated in that context BOs may live 3210be98a7SChris Wilson * longer than the context itself because of the way the hardware, and object 3310be98a7SChris Wilson * tracking works. Below is a very crude representation of the state machine 3410be98a7SChris Wilson * describing the context life. 3510be98a7SChris Wilson * refcount pincount active 3610be98a7SChris Wilson * S0: initial state 0 0 0 3710be98a7SChris Wilson * S1: context created 1 0 0 3810be98a7SChris Wilson * S2: context is currently running 2 1 X 3910be98a7SChris Wilson * S3: GPU referenced, but not current 2 0 1 4010be98a7SChris Wilson * S4: context is current, but destroyed 1 1 0 4110be98a7SChris Wilson * S5: like S3, but destroyed 1 0 1 4210be98a7SChris Wilson * 4310be98a7SChris Wilson * The most common (but not all) transitions: 4410be98a7SChris Wilson * S0->S1: client creates a context 4510be98a7SChris Wilson * S1->S2: client submits execbuf with context 4610be98a7SChris Wilson * S2->S3: other clients submits execbuf with context 4710be98a7SChris Wilson * S3->S1: context object was retired 4810be98a7SChris Wilson * S3->S2: clients submits another execbuf 4910be98a7SChris Wilson * S2->S4: context destroy called with current context 5010be98a7SChris Wilson * S3->S5->S0: destroy path 5110be98a7SChris Wilson * S4->S5->S0: destroy path on current context 5210be98a7SChris Wilson * 5310be98a7SChris Wilson * There are two confusing terms used above: 5410be98a7SChris Wilson * The "current context" means the context which is currently running on the 5510be98a7SChris Wilson * GPU. The GPU has loaded its state already and has stored away the gtt 5610be98a7SChris Wilson * offset of the BO. The GPU is not actively referencing the data at this 5710be98a7SChris Wilson * offset, but it will on the next context switch. The only way to avoid this 5810be98a7SChris Wilson * is to do a GPU reset. 5910be98a7SChris Wilson * 6010be98a7SChris Wilson * An "active context' is one which was previously the "current context" and is 6110be98a7SChris Wilson * on the active list waiting for the next context switch to occur. Until this 6210be98a7SChris Wilson * happens, the object must remain at the same gtt offset. It is therefore 6310be98a7SChris Wilson * possible to destroy a context, but it is still active. 6410be98a7SChris Wilson * 6510be98a7SChris Wilson */ 6610be98a7SChris Wilson 6710be98a7SChris Wilson #include <linux/log2.h> 6810be98a7SChris Wilson #include <linux/nospec.h> 6910be98a7SChris Wilson 7010be98a7SChris Wilson #include <drm/i915_drm.h> 7110be98a7SChris Wilson 722c86e55dSMatthew Auld #include "gt/gen6_ppgtt.h" 739f3ccd40SChris Wilson #include "gt/intel_context.h" 742e0986a5SChris Wilson #include "gt/intel_engine_heartbeat.h" 75750e76b4SChris Wilson #include "gt/intel_engine_user.h" 762871ea85SChris Wilson #include "gt/intel_ring.h" 7710be98a7SChris Wilson 7810be98a7SChris Wilson #include "i915_gem_context.h" 7910be98a7SChris Wilson #include "i915_globals.h" 8010be98a7SChris Wilson #include "i915_trace.h" 8110be98a7SChris Wilson #include "i915_user_extensions.h" 8210be98a7SChris Wilson 8310be98a7SChris Wilson #define ALL_L3_SLICES(dev) (1 << NUM_L3_SLICES(dev)) - 1 8410be98a7SChris Wilson 8510be98a7SChris Wilson static struct i915_global_gem_context { 8610be98a7SChris Wilson struct i915_global base; 8710be98a7SChris Wilson struct kmem_cache *slab_luts; 8810be98a7SChris Wilson } global; 8910be98a7SChris Wilson 9010be98a7SChris Wilson struct i915_lut_handle *i915_lut_handle_alloc(void) 9110be98a7SChris Wilson { 9210be98a7SChris Wilson return kmem_cache_alloc(global.slab_luts, GFP_KERNEL); 9310be98a7SChris Wilson } 9410be98a7SChris Wilson 9510be98a7SChris Wilson void i915_lut_handle_free(struct i915_lut_handle *lut) 9610be98a7SChris Wilson { 9710be98a7SChris Wilson return kmem_cache_free(global.slab_luts, lut); 9810be98a7SChris Wilson } 9910be98a7SChris Wilson 10010be98a7SChris Wilson static void lut_close(struct i915_gem_context *ctx) 10110be98a7SChris Wilson { 10210be98a7SChris Wilson struct radix_tree_iter iter; 10310be98a7SChris Wilson void __rcu **slot; 10410be98a7SChris Wilson 105155ab883SChris Wilson lockdep_assert_held(&ctx->mutex); 10610be98a7SChris Wilson 10710be98a7SChris Wilson rcu_read_lock(); 10810be98a7SChris Wilson radix_tree_for_each_slot(slot, &ctx->handles_vma, &iter, 0) { 10910be98a7SChris Wilson struct i915_vma *vma = rcu_dereference_raw(*slot); 110155ab883SChris Wilson struct drm_i915_gem_object *obj = vma->obj; 111155ab883SChris Wilson struct i915_lut_handle *lut; 11210be98a7SChris Wilson 113155ab883SChris Wilson if (!kref_get_unless_zero(&obj->base.refcount)) 114155ab883SChris Wilson continue; 115155ab883SChris Wilson 116155ab883SChris Wilson rcu_read_unlock(); 117155ab883SChris Wilson i915_gem_object_lock(obj); 118155ab883SChris Wilson list_for_each_entry(lut, &obj->lut_list, obj_link) { 119155ab883SChris Wilson if (lut->ctx != ctx) 120155ab883SChris Wilson continue; 121155ab883SChris Wilson 122155ab883SChris Wilson if (lut->handle != iter.index) 123155ab883SChris Wilson continue; 124155ab883SChris Wilson 125155ab883SChris Wilson list_del(&lut->obj_link); 126155ab883SChris Wilson break; 127155ab883SChris Wilson } 128155ab883SChris Wilson i915_gem_object_unlock(obj); 129155ab883SChris Wilson rcu_read_lock(); 130155ab883SChris Wilson 131155ab883SChris Wilson if (&lut->obj_link != &obj->lut_list) { 132155ab883SChris Wilson i915_lut_handle_free(lut); 13310be98a7SChris Wilson radix_tree_iter_delete(&ctx->handles_vma, &iter, slot); 134155ab883SChris Wilson if (atomic_dec_and_test(&vma->open_count) && 135155ab883SChris Wilson !i915_vma_is_ggtt(vma)) 136155ab883SChris Wilson i915_vma_close(vma); 137155ab883SChris Wilson i915_gem_object_put(obj); 138155ab883SChris Wilson } 13910be98a7SChris Wilson 140155ab883SChris Wilson i915_gem_object_put(obj); 14110be98a7SChris Wilson } 14210be98a7SChris Wilson rcu_read_unlock(); 14310be98a7SChris Wilson } 14410be98a7SChris Wilson 14510be98a7SChris Wilson static struct intel_context * 14610be98a7SChris Wilson lookup_user_engine(struct i915_gem_context *ctx, 14710be98a7SChris Wilson unsigned long flags, 14810be98a7SChris Wilson const struct i915_engine_class_instance *ci) 14910be98a7SChris Wilson #define LOOKUP_USER_INDEX BIT(0) 15010be98a7SChris Wilson { 15110be98a7SChris Wilson int idx; 15210be98a7SChris Wilson 15310be98a7SChris Wilson if (!!(flags & LOOKUP_USER_INDEX) != i915_gem_context_user_engines(ctx)) 15410be98a7SChris Wilson return ERR_PTR(-EINVAL); 15510be98a7SChris Wilson 15610be98a7SChris Wilson if (!i915_gem_context_user_engines(ctx)) { 15710be98a7SChris Wilson struct intel_engine_cs *engine; 15810be98a7SChris Wilson 15910be98a7SChris Wilson engine = intel_engine_lookup_user(ctx->i915, 16010be98a7SChris Wilson ci->engine_class, 16110be98a7SChris Wilson ci->engine_instance); 16210be98a7SChris Wilson if (!engine) 16310be98a7SChris Wilson return ERR_PTR(-EINVAL); 16410be98a7SChris Wilson 165f1c4d157SChris Wilson idx = engine->legacy_idx; 16610be98a7SChris Wilson } else { 16710be98a7SChris Wilson idx = ci->engine_instance; 16810be98a7SChris Wilson } 16910be98a7SChris Wilson 17010be98a7SChris Wilson return i915_gem_context_get_engine(ctx, idx); 17110be98a7SChris Wilson } 17210be98a7SChris Wilson 17327dbae8fSChris Wilson static struct i915_address_space * 17427dbae8fSChris Wilson context_get_vm_rcu(struct i915_gem_context *ctx) 17527dbae8fSChris Wilson { 17627dbae8fSChris Wilson GEM_BUG_ON(!rcu_access_pointer(ctx->vm)); 17727dbae8fSChris Wilson 17827dbae8fSChris Wilson do { 17927dbae8fSChris Wilson struct i915_address_space *vm; 18027dbae8fSChris Wilson 18127dbae8fSChris Wilson /* 18227dbae8fSChris Wilson * We do not allow downgrading from full-ppgtt [to a shared 18327dbae8fSChris Wilson * global gtt], so ctx->vm cannot become NULL. 18427dbae8fSChris Wilson */ 18527dbae8fSChris Wilson vm = rcu_dereference(ctx->vm); 18627dbae8fSChris Wilson if (!kref_get_unless_zero(&vm->ref)) 18727dbae8fSChris Wilson continue; 18827dbae8fSChris Wilson 18927dbae8fSChris Wilson /* 19027dbae8fSChris Wilson * This ppgtt may have be reallocated between 19127dbae8fSChris Wilson * the read and the kref, and reassigned to a third 19227dbae8fSChris Wilson * context. In order to avoid inadvertent sharing 19327dbae8fSChris Wilson * of this ppgtt with that third context (and not 19427dbae8fSChris Wilson * src), we have to confirm that we have the same 19527dbae8fSChris Wilson * ppgtt after passing through the strong memory 19627dbae8fSChris Wilson * barrier implied by a successful 19727dbae8fSChris Wilson * kref_get_unless_zero(). 19827dbae8fSChris Wilson * 19927dbae8fSChris Wilson * Once we have acquired the current ppgtt of ctx, 20027dbae8fSChris Wilson * we no longer care if it is released from ctx, as 20127dbae8fSChris Wilson * it cannot be reallocated elsewhere. 20227dbae8fSChris Wilson */ 20327dbae8fSChris Wilson 20427dbae8fSChris Wilson if (vm == rcu_access_pointer(ctx->vm)) 20527dbae8fSChris Wilson return rcu_pointer_handoff(vm); 20627dbae8fSChris Wilson 20727dbae8fSChris Wilson i915_vm_put(vm); 20827dbae8fSChris Wilson } while (1); 20927dbae8fSChris Wilson } 21027dbae8fSChris Wilson 211e6ba7648SChris Wilson static void intel_context_set_gem(struct intel_context *ce, 212e6ba7648SChris Wilson struct i915_gem_context *ctx) 213e6ba7648SChris Wilson { 2146a8679c0SChris Wilson GEM_BUG_ON(rcu_access_pointer(ce->gem_context)); 2156a8679c0SChris Wilson RCU_INIT_POINTER(ce->gem_context, ctx); 216e6ba7648SChris Wilson 217e6ba7648SChris Wilson if (!test_bit(CONTEXT_ALLOC_BIT, &ce->flags)) 218e6ba7648SChris Wilson ce->ring = __intel_context_ring_size(SZ_16K); 219e6ba7648SChris Wilson 220e6ba7648SChris Wilson if (rcu_access_pointer(ctx->vm)) { 221e6ba7648SChris Wilson struct i915_address_space *vm; 222e6ba7648SChris Wilson 223e6ba7648SChris Wilson rcu_read_lock(); 224e6ba7648SChris Wilson vm = context_get_vm_rcu(ctx); /* hmm */ 225e6ba7648SChris Wilson rcu_read_unlock(); 226e6ba7648SChris Wilson 227e6ba7648SChris Wilson i915_vm_put(ce->vm); 228e6ba7648SChris Wilson ce->vm = vm; 229e6ba7648SChris Wilson } 230e6ba7648SChris Wilson 231e6ba7648SChris Wilson GEM_BUG_ON(ce->timeline); 232e6ba7648SChris Wilson if (ctx->timeline) 233e6ba7648SChris Wilson ce->timeline = intel_timeline_get(ctx->timeline); 234e6ba7648SChris Wilson 235e6ba7648SChris Wilson if (ctx->sched.priority >= I915_PRIORITY_NORMAL && 236e6ba7648SChris Wilson intel_engine_has_semaphores(ce->engine)) 237e6ba7648SChris Wilson __set_bit(CONTEXT_USE_SEMAPHORES, &ce->flags); 238e6ba7648SChris Wilson } 239e6ba7648SChris Wilson 24010be98a7SChris Wilson static void __free_engines(struct i915_gem_engines *e, unsigned int count) 24110be98a7SChris Wilson { 24210be98a7SChris Wilson while (count--) { 24310be98a7SChris Wilson if (!e->engines[count]) 24410be98a7SChris Wilson continue; 24510be98a7SChris Wilson 2466a8679c0SChris Wilson RCU_INIT_POINTER(e->engines[count]->gem_context, NULL); 24710be98a7SChris Wilson intel_context_put(e->engines[count]); 24810be98a7SChris Wilson } 24910be98a7SChris Wilson kfree(e); 25010be98a7SChris Wilson } 25110be98a7SChris Wilson 25210be98a7SChris Wilson static void free_engines(struct i915_gem_engines *e) 25310be98a7SChris Wilson { 25410be98a7SChris Wilson __free_engines(e, e->num_engines); 25510be98a7SChris Wilson } 25610be98a7SChris Wilson 257155ab883SChris Wilson static void free_engines_rcu(struct rcu_head *rcu) 25810be98a7SChris Wilson { 259155ab883SChris Wilson free_engines(container_of(rcu, struct i915_gem_engines, rcu)); 26010be98a7SChris Wilson } 26110be98a7SChris Wilson 26210be98a7SChris Wilson static struct i915_gem_engines *default_engines(struct i915_gem_context *ctx) 26310be98a7SChris Wilson { 264f1c4d157SChris Wilson const struct intel_gt *gt = &ctx->i915->gt; 26510be98a7SChris Wilson struct intel_engine_cs *engine; 26610be98a7SChris Wilson struct i915_gem_engines *e; 26710be98a7SChris Wilson enum intel_engine_id id; 26810be98a7SChris Wilson 26910be98a7SChris Wilson e = kzalloc(struct_size(e, engines, I915_NUM_ENGINES), GFP_KERNEL); 27010be98a7SChris Wilson if (!e) 27110be98a7SChris Wilson return ERR_PTR(-ENOMEM); 27210be98a7SChris Wilson 273155ab883SChris Wilson init_rcu_head(&e->rcu); 274f1c4d157SChris Wilson for_each_engine(engine, gt, id) { 27510be98a7SChris Wilson struct intel_context *ce; 27610be98a7SChris Wilson 277a50134b1STvrtko Ursulin if (engine->legacy_idx == INVALID_ENGINE) 278a50134b1STvrtko Ursulin continue; 279a50134b1STvrtko Ursulin 280a50134b1STvrtko Ursulin GEM_BUG_ON(engine->legacy_idx >= I915_NUM_ENGINES); 281a50134b1STvrtko Ursulin GEM_BUG_ON(e->engines[engine->legacy_idx]); 282a50134b1STvrtko Ursulin 283e6ba7648SChris Wilson ce = intel_context_create(engine); 28410be98a7SChris Wilson if (IS_ERR(ce)) { 285a50134b1STvrtko Ursulin __free_engines(e, e->num_engines + 1); 28610be98a7SChris Wilson return ERR_CAST(ce); 28710be98a7SChris Wilson } 28810be98a7SChris Wilson 289e6ba7648SChris Wilson intel_context_set_gem(ce, ctx); 290e6ba7648SChris Wilson 291a50134b1STvrtko Ursulin e->engines[engine->legacy_idx] = ce; 292a50134b1STvrtko Ursulin e->num_engines = max(e->num_engines, engine->legacy_idx); 29310be98a7SChris Wilson } 294a50134b1STvrtko Ursulin e->num_engines++; 29510be98a7SChris Wilson 29610be98a7SChris Wilson return e; 29710be98a7SChris Wilson } 29810be98a7SChris Wilson 29910be98a7SChris Wilson static void i915_gem_context_free(struct i915_gem_context *ctx) 30010be98a7SChris Wilson { 30110be98a7SChris Wilson GEM_BUG_ON(!i915_gem_context_is_closed(ctx)); 30210be98a7SChris Wilson 303a4e7ccdaSChris Wilson spin_lock(&ctx->i915->gem.contexts.lock); 304a4e7ccdaSChris Wilson list_del(&ctx->link); 305a4e7ccdaSChris Wilson spin_unlock(&ctx->i915->gem.contexts.lock); 306a4e7ccdaSChris Wilson 30710be98a7SChris Wilson free_engines(rcu_access_pointer(ctx->engines)); 30810be98a7SChris Wilson mutex_destroy(&ctx->engines_mutex); 30910be98a7SChris Wilson 31010be98a7SChris Wilson if (ctx->timeline) 311f0c02c1bSTvrtko Ursulin intel_timeline_put(ctx->timeline); 31210be98a7SChris Wilson 31310be98a7SChris Wilson put_pid(ctx->pid); 31410be98a7SChris Wilson mutex_destroy(&ctx->mutex); 31510be98a7SChris Wilson 31610be98a7SChris Wilson kfree_rcu(ctx, rcu); 31710be98a7SChris Wilson } 31810be98a7SChris Wilson 319a4e7ccdaSChris Wilson static void contexts_free_all(struct llist_node *list) 32010be98a7SChris Wilson { 32110be98a7SChris Wilson struct i915_gem_context *ctx, *cn; 32210be98a7SChris Wilson 323a4e7ccdaSChris Wilson llist_for_each_entry_safe(ctx, cn, list, free_link) 32410be98a7SChris Wilson i915_gem_context_free(ctx); 32510be98a7SChris Wilson } 32610be98a7SChris Wilson 327a4e7ccdaSChris Wilson static void contexts_flush_free(struct i915_gem_contexts *gc) 32810be98a7SChris Wilson { 329a4e7ccdaSChris Wilson contexts_free_all(llist_del_all(&gc->free_list)); 33010be98a7SChris Wilson } 33110be98a7SChris Wilson 33210be98a7SChris Wilson static void contexts_free_worker(struct work_struct *work) 33310be98a7SChris Wilson { 334a4e7ccdaSChris Wilson struct i915_gem_contexts *gc = 335a4e7ccdaSChris Wilson container_of(work, typeof(*gc), free_work); 33610be98a7SChris Wilson 337a4e7ccdaSChris Wilson contexts_flush_free(gc); 33810be98a7SChris Wilson } 33910be98a7SChris Wilson 34010be98a7SChris Wilson void i915_gem_context_release(struct kref *ref) 34110be98a7SChris Wilson { 34210be98a7SChris Wilson struct i915_gem_context *ctx = container_of(ref, typeof(*ctx), ref); 343a4e7ccdaSChris Wilson struct i915_gem_contexts *gc = &ctx->i915->gem.contexts; 34410be98a7SChris Wilson 34510be98a7SChris Wilson trace_i915_context_free(ctx); 346a4e7ccdaSChris Wilson if (llist_add(&ctx->free_link, &gc->free_list)) 347a4e7ccdaSChris Wilson schedule_work(&gc->free_work); 34810be98a7SChris Wilson } 34910be98a7SChris Wilson 3502e0986a5SChris Wilson static inline struct i915_gem_engines * 3512e0986a5SChris Wilson __context_engines_static(const struct i915_gem_context *ctx) 3522e0986a5SChris Wilson { 3532e0986a5SChris Wilson return rcu_dereference_protected(ctx->engines, true); 3542e0986a5SChris Wilson } 3552e0986a5SChris Wilson 3562e0986a5SChris Wilson static bool __reset_engine(struct intel_engine_cs *engine) 3572e0986a5SChris Wilson { 3582e0986a5SChris Wilson struct intel_gt *gt = engine->gt; 3592e0986a5SChris Wilson bool success = false; 3602e0986a5SChris Wilson 3612e0986a5SChris Wilson if (!intel_has_reset_engine(gt)) 3622e0986a5SChris Wilson return false; 3632e0986a5SChris Wilson 3642e0986a5SChris Wilson if (!test_and_set_bit(I915_RESET_ENGINE + engine->id, 3652e0986a5SChris Wilson >->reset.flags)) { 3662e0986a5SChris Wilson success = intel_engine_reset(engine, NULL) == 0; 3672e0986a5SChris Wilson clear_and_wake_up_bit(I915_RESET_ENGINE + engine->id, 3682e0986a5SChris Wilson >->reset.flags); 3692e0986a5SChris Wilson } 3702e0986a5SChris Wilson 3712e0986a5SChris Wilson return success; 3722e0986a5SChris Wilson } 3732e0986a5SChris Wilson 3742e0986a5SChris Wilson static void __reset_context(struct i915_gem_context *ctx, 3752e0986a5SChris Wilson struct intel_engine_cs *engine) 3762e0986a5SChris Wilson { 3772e0986a5SChris Wilson intel_gt_handle_error(engine->gt, engine->mask, 0, 3782e0986a5SChris Wilson "context closure in %s", ctx->name); 3792e0986a5SChris Wilson } 3802e0986a5SChris Wilson 3812e0986a5SChris Wilson static bool __cancel_engine(struct intel_engine_cs *engine) 3822e0986a5SChris Wilson { 3832e0986a5SChris Wilson /* 3842e0986a5SChris Wilson * Send a "high priority pulse" down the engine to cause the 3852e0986a5SChris Wilson * current request to be momentarily preempted. (If it fails to 3862e0986a5SChris Wilson * be preempted, it will be reset). As we have marked our context 3872e0986a5SChris Wilson * as banned, any incomplete request, including any running, will 3882e0986a5SChris Wilson * be skipped following the preemption. 3892e0986a5SChris Wilson * 3902e0986a5SChris Wilson * If there is no hangchecking (one of the reasons why we try to 3912e0986a5SChris Wilson * cancel the context) and no forced preemption, there may be no 3922e0986a5SChris Wilson * means by which we reset the GPU and evict the persistent hog. 3932e0986a5SChris Wilson * Ergo if we are unable to inject a preemptive pulse that can 3942e0986a5SChris Wilson * kill the banned context, we fallback to doing a local reset 3952e0986a5SChris Wilson * instead. 3962e0986a5SChris Wilson */ 397babaab2fSChris Wilson if (IS_ACTIVE(CONFIG_DRM_I915_PREEMPT_TIMEOUT) && 398babaab2fSChris Wilson !intel_engine_pulse(engine)) 3992e0986a5SChris Wilson return true; 4002e0986a5SChris Wilson 4012e0986a5SChris Wilson /* If we are unable to send a pulse, try resetting this engine. */ 4022e0986a5SChris Wilson return __reset_engine(engine); 4032e0986a5SChris Wilson } 4042e0986a5SChris Wilson 4054a317415SChris Wilson static struct intel_engine_cs *__active_engine(struct i915_request *rq) 4062e0986a5SChris Wilson { 4072e0986a5SChris Wilson struct intel_engine_cs *engine, *locked; 4082e0986a5SChris Wilson 4092e0986a5SChris Wilson /* 4102e0986a5SChris Wilson * Serialise with __i915_request_submit() so that it sees 4112e0986a5SChris Wilson * is-banned?, or we know the request is already inflight. 4122e0986a5SChris Wilson */ 4132e0986a5SChris Wilson locked = READ_ONCE(rq->engine); 4142e0986a5SChris Wilson spin_lock_irq(&locked->active.lock); 4152e0986a5SChris Wilson while (unlikely(locked != (engine = READ_ONCE(rq->engine)))) { 4162e0986a5SChris Wilson spin_unlock(&locked->active.lock); 4172e0986a5SChris Wilson spin_lock(&engine->active.lock); 4182e0986a5SChris Wilson locked = engine; 4192e0986a5SChris Wilson } 4202e0986a5SChris Wilson 4212e0986a5SChris Wilson engine = NULL; 4226e17ae73SChris Wilson if (i915_request_is_active(rq) && rq->fence.error != -EIO) 4232e0986a5SChris Wilson engine = rq->engine; 4242e0986a5SChris Wilson 4252e0986a5SChris Wilson spin_unlock_irq(&locked->active.lock); 4262e0986a5SChris Wilson 4272e0986a5SChris Wilson return engine; 4282e0986a5SChris Wilson } 4292e0986a5SChris Wilson 4304a317415SChris Wilson static struct intel_engine_cs *active_engine(struct intel_context *ce) 4314a317415SChris Wilson { 4324a317415SChris Wilson struct intel_engine_cs *engine = NULL; 4334a317415SChris Wilson struct i915_request *rq; 4344a317415SChris Wilson 4354a317415SChris Wilson if (!ce->timeline) 4364a317415SChris Wilson return NULL; 4374a317415SChris Wilson 4387ce596a8SChris Wilson mutex_lock(&ce->timeline->mutex); 4394a317415SChris Wilson list_for_each_entry_reverse(rq, &ce->timeline->requests, link) { 4404a317415SChris Wilson if (i915_request_completed(rq)) 4414a317415SChris Wilson break; 4424a317415SChris Wilson 4434a317415SChris Wilson /* Check with the backend if the request is inflight */ 4444a317415SChris Wilson engine = __active_engine(rq); 4454a317415SChris Wilson if (engine) 4464a317415SChris Wilson break; 4474a317415SChris Wilson } 4487ce596a8SChris Wilson mutex_unlock(&ce->timeline->mutex); 4494a317415SChris Wilson 4504a317415SChris Wilson return engine; 4514a317415SChris Wilson } 4524a317415SChris Wilson 4532e0986a5SChris Wilson static void kill_context(struct i915_gem_context *ctx) 4542e0986a5SChris Wilson { 4552e0986a5SChris Wilson struct i915_gem_engines_iter it; 4562e0986a5SChris Wilson struct intel_context *ce; 4572e0986a5SChris Wilson 4582e0986a5SChris Wilson /* 4592e0986a5SChris Wilson * Map the user's engine back to the actual engines; one virtual 4602e0986a5SChris Wilson * engine will be mapped to multiple engines, and using ctx->engine[] 4612e0986a5SChris Wilson * the same engine may be have multiple instances in the user's map. 4622e0986a5SChris Wilson * However, we only care about pending requests, so only include 4632e0986a5SChris Wilson * engines on which there are incomplete requests. 4642e0986a5SChris Wilson */ 4652e0986a5SChris Wilson for_each_gem_engine(ce, __context_engines_static(ctx), it) { 4662e0986a5SChris Wilson struct intel_engine_cs *engine; 4672e0986a5SChris Wilson 4689f3ccd40SChris Wilson if (intel_context_set_banned(ce)) 4699f3ccd40SChris Wilson continue; 4709f3ccd40SChris Wilson 4714a317415SChris Wilson /* 4724a317415SChris Wilson * Check the current active state of this context; if we 4734a317415SChris Wilson * are currently executing on the GPU we need to evict 4744a317415SChris Wilson * ourselves. On the other hand, if we haven't yet been 4754a317415SChris Wilson * submitted to the GPU or if everything is complete, 4764a317415SChris Wilson * we have nothing to do. 4774a317415SChris Wilson */ 4784a317415SChris Wilson engine = active_engine(ce); 4792e0986a5SChris Wilson 4802e0986a5SChris Wilson /* First attempt to gracefully cancel the context */ 4812e0986a5SChris Wilson if (engine && !__cancel_engine(engine)) 4822e0986a5SChris Wilson /* 4832e0986a5SChris Wilson * If we are unable to send a preemptive pulse to bump 4842e0986a5SChris Wilson * the context from the GPU, we have to resort to a full 4852e0986a5SChris Wilson * reset. We hope the collateral damage is worth it. 4862e0986a5SChris Wilson */ 4872e0986a5SChris Wilson __reset_context(ctx, engine); 4882e0986a5SChris Wilson } 4892e0986a5SChris Wilson } 4902e0986a5SChris Wilson 491267c0126SChris Wilson static void set_closed_name(struct i915_gem_context *ctx) 492267c0126SChris Wilson { 493267c0126SChris Wilson char *s; 494267c0126SChris Wilson 495267c0126SChris Wilson /* Replace '[]' with '<>' to indicate closed in debug prints */ 496267c0126SChris Wilson 497267c0126SChris Wilson s = strrchr(ctx->name, '['); 498267c0126SChris Wilson if (!s) 499267c0126SChris Wilson return; 500267c0126SChris Wilson 501267c0126SChris Wilson *s = '<'; 502267c0126SChris Wilson 503267c0126SChris Wilson s = strchr(s + 1, ']'); 504267c0126SChris Wilson if (s) 505267c0126SChris Wilson *s = '>'; 506267c0126SChris Wilson } 507267c0126SChris Wilson 50810be98a7SChris Wilson static void context_close(struct i915_gem_context *ctx) 50910be98a7SChris Wilson { 510a4e7ccdaSChris Wilson struct i915_address_space *vm; 511a4e7ccdaSChris Wilson 5122850748eSChris Wilson i915_gem_context_set_closed(ctx); 513267c0126SChris Wilson set_closed_name(ctx); 5142850748eSChris Wilson 515155ab883SChris Wilson mutex_lock(&ctx->mutex); 516155ab883SChris Wilson 517a4e7ccdaSChris Wilson vm = i915_gem_context_vm(ctx); 518a4e7ccdaSChris Wilson if (vm) 519a4e7ccdaSChris Wilson i915_vm_close(vm); 520a4e7ccdaSChris Wilson 521155ab883SChris Wilson ctx->file_priv = ERR_PTR(-EBADF); 52210be98a7SChris Wilson 52310be98a7SChris Wilson /* 52410be98a7SChris Wilson * The LUT uses the VMA as a backpointer to unref the object, 52510be98a7SChris Wilson * so we need to clear the LUT before we close all the VMA (inside 52610be98a7SChris Wilson * the ppgtt). 52710be98a7SChris Wilson */ 52810be98a7SChris Wilson lut_close(ctx); 52910be98a7SChris Wilson 530155ab883SChris Wilson mutex_unlock(&ctx->mutex); 5312e0986a5SChris Wilson 5322e0986a5SChris Wilson /* 5332e0986a5SChris Wilson * If the user has disabled hangchecking, we can not be sure that 5342e0986a5SChris Wilson * the batches will ever complete after the context is closed, 5352e0986a5SChris Wilson * keeping the context and all resources pinned forever. So in this 5362e0986a5SChris Wilson * case we opt to forcibly kill off all remaining requests on 5372e0986a5SChris Wilson * context close. 5382e0986a5SChris Wilson */ 539a0e04715SChris Wilson if (!i915_gem_context_is_persistent(ctx) || 540a0e04715SChris Wilson !i915_modparams.enable_hangcheck) 5412e0986a5SChris Wilson kill_context(ctx); 5422e0986a5SChris Wilson 54310be98a7SChris Wilson i915_gem_context_put(ctx); 54410be98a7SChris Wilson } 54510be98a7SChris Wilson 546a0e04715SChris Wilson static int __context_set_persistence(struct i915_gem_context *ctx, bool state) 547a0e04715SChris Wilson { 548a0e04715SChris Wilson if (i915_gem_context_is_persistent(ctx) == state) 549a0e04715SChris Wilson return 0; 550a0e04715SChris Wilson 551a0e04715SChris Wilson if (state) { 552a0e04715SChris Wilson /* 553a0e04715SChris Wilson * Only contexts that are short-lived [that will expire or be 554a0e04715SChris Wilson * reset] are allowed to survive past termination. We require 555a0e04715SChris Wilson * hangcheck to ensure that the persistent requests are healthy. 556a0e04715SChris Wilson */ 557a0e04715SChris Wilson if (!i915_modparams.enable_hangcheck) 558a0e04715SChris Wilson return -EINVAL; 559a0e04715SChris Wilson 560a0e04715SChris Wilson i915_gem_context_set_persistence(ctx); 561a0e04715SChris Wilson } else { 562a0e04715SChris Wilson /* To cancel a context we use "preempt-to-idle" */ 563a0e04715SChris Wilson if (!(ctx->i915->caps.scheduler & I915_SCHEDULER_CAP_PREEMPTION)) 564a0e04715SChris Wilson return -ENODEV; 565a0e04715SChris Wilson 566d1b9b5f1SChris Wilson /* 567d1b9b5f1SChris Wilson * If the cancel fails, we then need to reset, cleanly! 568d1b9b5f1SChris Wilson * 569d1b9b5f1SChris Wilson * If the per-engine reset fails, all hope is lost! We resort 570d1b9b5f1SChris Wilson * to a full GPU reset in that unlikely case, but realistically 571d1b9b5f1SChris Wilson * if the engine could not reset, the full reset does not fare 572d1b9b5f1SChris Wilson * much better. The damage has been done. 573d1b9b5f1SChris Wilson * 574d1b9b5f1SChris Wilson * However, if we cannot reset an engine by itself, we cannot 575d1b9b5f1SChris Wilson * cleanup a hanging persistent context without causing 576d1b9b5f1SChris Wilson * colateral damage, and we should not pretend we can by 577d1b9b5f1SChris Wilson * exposing the interface. 578d1b9b5f1SChris Wilson */ 579d1b9b5f1SChris Wilson if (!intel_has_reset_engine(&ctx->i915->gt)) 580d1b9b5f1SChris Wilson return -ENODEV; 581d1b9b5f1SChris Wilson 582a0e04715SChris Wilson i915_gem_context_clear_persistence(ctx); 583a0e04715SChris Wilson } 584a0e04715SChris Wilson 585a0e04715SChris Wilson return 0; 586a0e04715SChris Wilson } 587a0e04715SChris Wilson 58810be98a7SChris Wilson static struct i915_gem_context * 589e568ac38SChris Wilson __create_context(struct drm_i915_private *i915) 59010be98a7SChris Wilson { 59110be98a7SChris Wilson struct i915_gem_context *ctx; 59210be98a7SChris Wilson struct i915_gem_engines *e; 59310be98a7SChris Wilson int err; 59410be98a7SChris Wilson int i; 59510be98a7SChris Wilson 59610be98a7SChris Wilson ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 59710be98a7SChris Wilson if (!ctx) 59810be98a7SChris Wilson return ERR_PTR(-ENOMEM); 59910be98a7SChris Wilson 60010be98a7SChris Wilson kref_init(&ctx->ref); 601e568ac38SChris Wilson ctx->i915 = i915; 60210be98a7SChris Wilson ctx->sched.priority = I915_USER_PRIORITY(I915_PRIORITY_NORMAL); 60310be98a7SChris Wilson mutex_init(&ctx->mutex); 60410be98a7SChris Wilson 60510be98a7SChris Wilson mutex_init(&ctx->engines_mutex); 60610be98a7SChris Wilson e = default_engines(ctx); 60710be98a7SChris Wilson if (IS_ERR(e)) { 60810be98a7SChris Wilson err = PTR_ERR(e); 60910be98a7SChris Wilson goto err_free; 61010be98a7SChris Wilson } 61110be98a7SChris Wilson RCU_INIT_POINTER(ctx->engines, e); 61210be98a7SChris Wilson 61310be98a7SChris Wilson INIT_RADIX_TREE(&ctx->handles_vma, GFP_KERNEL); 61410be98a7SChris Wilson 61510be98a7SChris Wilson /* NB: Mark all slices as needing a remap so that when the context first 61610be98a7SChris Wilson * loads it will restore whatever remap state already exists. If there 61710be98a7SChris Wilson * is no remap info, it will be a NOP. */ 618e568ac38SChris Wilson ctx->remap_slice = ALL_L3_SLICES(i915); 61910be98a7SChris Wilson 62010be98a7SChris Wilson i915_gem_context_set_bannable(ctx); 62110be98a7SChris Wilson i915_gem_context_set_recoverable(ctx); 622a0e04715SChris Wilson __context_set_persistence(ctx, true /* cgroup hook? */); 62310be98a7SChris Wilson 62410be98a7SChris Wilson for (i = 0; i < ARRAY_SIZE(ctx->hang_timestamp); i++) 62510be98a7SChris Wilson ctx->hang_timestamp[i] = jiffies - CONTEXT_FAST_HANG_JIFFIES; 62610be98a7SChris Wilson 627a4e7ccdaSChris Wilson spin_lock(&i915->gem.contexts.lock); 628a4e7ccdaSChris Wilson list_add_tail(&ctx->link, &i915->gem.contexts.list); 629a4e7ccdaSChris Wilson spin_unlock(&i915->gem.contexts.lock); 630a4e7ccdaSChris Wilson 63110be98a7SChris Wilson return ctx; 63210be98a7SChris Wilson 63310be98a7SChris Wilson err_free: 63410be98a7SChris Wilson kfree(ctx); 63510be98a7SChris Wilson return ERR_PTR(err); 63610be98a7SChris Wilson } 63710be98a7SChris Wilson 63848ae397bSChris Wilson static void 63948ae397bSChris Wilson context_apply_all(struct i915_gem_context *ctx, 64048ae397bSChris Wilson void (*fn)(struct intel_context *ce, void *data), 64148ae397bSChris Wilson void *data) 64248ae397bSChris Wilson { 64348ae397bSChris Wilson struct i915_gem_engines_iter it; 64448ae397bSChris Wilson struct intel_context *ce; 64548ae397bSChris Wilson 64648ae397bSChris Wilson for_each_gem_engine(ce, i915_gem_context_lock_engines(ctx), it) 64748ae397bSChris Wilson fn(ce, data); 64848ae397bSChris Wilson i915_gem_context_unlock_engines(ctx); 64948ae397bSChris Wilson } 65048ae397bSChris Wilson 65148ae397bSChris Wilson static void __apply_ppgtt(struct intel_context *ce, void *vm) 65248ae397bSChris Wilson { 65348ae397bSChris Wilson i915_vm_put(ce->vm); 65448ae397bSChris Wilson ce->vm = i915_vm_get(vm); 65548ae397bSChris Wilson } 65648ae397bSChris Wilson 657e568ac38SChris Wilson static struct i915_address_space * 658e568ac38SChris Wilson __set_ppgtt(struct i915_gem_context *ctx, struct i915_address_space *vm) 65910be98a7SChris Wilson { 660a4e7ccdaSChris Wilson struct i915_address_space *old = i915_gem_context_vm(ctx); 66110be98a7SChris Wilson 662a1c9ca22SChris Wilson GEM_BUG_ON(old && i915_vm_is_4lvl(vm) != i915_vm_is_4lvl(old)); 663a1c9ca22SChris Wilson 664a4e7ccdaSChris Wilson rcu_assign_pointer(ctx->vm, i915_vm_open(vm)); 66548ae397bSChris Wilson context_apply_all(ctx, __apply_ppgtt, vm); 666f5d974f9SChris Wilson 66710be98a7SChris Wilson return old; 66810be98a7SChris Wilson } 66910be98a7SChris Wilson 67010be98a7SChris Wilson static void __assign_ppgtt(struct i915_gem_context *ctx, 671e568ac38SChris Wilson struct i915_address_space *vm) 67210be98a7SChris Wilson { 673a4e7ccdaSChris Wilson if (vm == rcu_access_pointer(ctx->vm)) 67410be98a7SChris Wilson return; 67510be98a7SChris Wilson 676e568ac38SChris Wilson vm = __set_ppgtt(ctx, vm); 677e568ac38SChris Wilson if (vm) 6782850748eSChris Wilson i915_vm_close(vm); 67910be98a7SChris Wilson } 68010be98a7SChris Wilson 68175d0a7f3SChris Wilson static void __set_timeline(struct intel_timeline **dst, 68275d0a7f3SChris Wilson struct intel_timeline *src) 68375d0a7f3SChris Wilson { 68475d0a7f3SChris Wilson struct intel_timeline *old = *dst; 68575d0a7f3SChris Wilson 68675d0a7f3SChris Wilson *dst = src ? intel_timeline_get(src) : NULL; 68775d0a7f3SChris Wilson 68875d0a7f3SChris Wilson if (old) 68975d0a7f3SChris Wilson intel_timeline_put(old); 69075d0a7f3SChris Wilson } 69175d0a7f3SChris Wilson 69275d0a7f3SChris Wilson static void __apply_timeline(struct intel_context *ce, void *timeline) 69375d0a7f3SChris Wilson { 69475d0a7f3SChris Wilson __set_timeline(&ce->timeline, timeline); 69575d0a7f3SChris Wilson } 69675d0a7f3SChris Wilson 69775d0a7f3SChris Wilson static void __assign_timeline(struct i915_gem_context *ctx, 69875d0a7f3SChris Wilson struct intel_timeline *timeline) 69975d0a7f3SChris Wilson { 70075d0a7f3SChris Wilson __set_timeline(&ctx->timeline, timeline); 70175d0a7f3SChris Wilson context_apply_all(ctx, __apply_timeline, timeline); 70275d0a7f3SChris Wilson } 70375d0a7f3SChris Wilson 70410be98a7SChris Wilson static struct i915_gem_context * 705a4e7ccdaSChris Wilson i915_gem_create_context(struct drm_i915_private *i915, unsigned int flags) 70610be98a7SChris Wilson { 70710be98a7SChris Wilson struct i915_gem_context *ctx; 70810be98a7SChris Wilson 70910be98a7SChris Wilson if (flags & I915_CONTEXT_CREATE_FLAGS_SINGLE_TIMELINE && 710a4e7ccdaSChris Wilson !HAS_EXECLISTS(i915)) 71110be98a7SChris Wilson return ERR_PTR(-EINVAL); 71210be98a7SChris Wilson 713a4e7ccdaSChris Wilson /* Reap the stale contexts */ 714a4e7ccdaSChris Wilson contexts_flush_free(&i915->gem.contexts); 71510be98a7SChris Wilson 716a4e7ccdaSChris Wilson ctx = __create_context(i915); 71710be98a7SChris Wilson if (IS_ERR(ctx)) 71810be98a7SChris Wilson return ctx; 71910be98a7SChris Wilson 720a4e7ccdaSChris Wilson if (HAS_FULL_PPGTT(i915)) { 721ab53497bSChris Wilson struct i915_ppgtt *ppgtt; 72210be98a7SChris Wilson 7232c86e55dSMatthew Auld ppgtt = i915_ppgtt_create(&i915->gt); 72410be98a7SChris Wilson if (IS_ERR(ppgtt)) { 725baa89ba3SWambui Karuga drm_dbg(&i915->drm, "PPGTT setup failed (%ld)\n", 72610be98a7SChris Wilson PTR_ERR(ppgtt)); 72710be98a7SChris Wilson context_close(ctx); 72810be98a7SChris Wilson return ERR_CAST(ppgtt); 72910be98a7SChris Wilson } 73010be98a7SChris Wilson 731a4e7ccdaSChris Wilson mutex_lock(&ctx->mutex); 732e568ac38SChris Wilson __assign_ppgtt(ctx, &ppgtt->vm); 733a4e7ccdaSChris Wilson mutex_unlock(&ctx->mutex); 734a4e7ccdaSChris Wilson 735e568ac38SChris Wilson i915_vm_put(&ppgtt->vm); 73610be98a7SChris Wilson } 73710be98a7SChris Wilson 73810be98a7SChris Wilson if (flags & I915_CONTEXT_CREATE_FLAGS_SINGLE_TIMELINE) { 739f0c02c1bSTvrtko Ursulin struct intel_timeline *timeline; 74010be98a7SChris Wilson 741a4e7ccdaSChris Wilson timeline = intel_timeline_create(&i915->gt, NULL); 74210be98a7SChris Wilson if (IS_ERR(timeline)) { 74310be98a7SChris Wilson context_close(ctx); 74410be98a7SChris Wilson return ERR_CAST(timeline); 74510be98a7SChris Wilson } 74610be98a7SChris Wilson 74775d0a7f3SChris Wilson __assign_timeline(ctx, timeline); 74875d0a7f3SChris Wilson intel_timeline_put(timeline); 74910be98a7SChris Wilson } 75010be98a7SChris Wilson 75110be98a7SChris Wilson trace_i915_context_create(ctx); 75210be98a7SChris Wilson 75310be98a7SChris Wilson return ctx; 75410be98a7SChris Wilson } 75510be98a7SChris Wilson 756a4e7ccdaSChris Wilson static void init_contexts(struct i915_gem_contexts *gc) 75710be98a7SChris Wilson { 758a4e7ccdaSChris Wilson spin_lock_init(&gc->lock); 759a4e7ccdaSChris Wilson INIT_LIST_HEAD(&gc->list); 76010be98a7SChris Wilson 761a4e7ccdaSChris Wilson INIT_WORK(&gc->free_work, contexts_free_worker); 762a4e7ccdaSChris Wilson init_llist_head(&gc->free_list); 76310be98a7SChris Wilson } 76410be98a7SChris Wilson 765e6ba7648SChris Wilson void i915_gem_init__contexts(struct drm_i915_private *i915) 76610be98a7SChris Wilson { 767a4e7ccdaSChris Wilson init_contexts(&i915->gem.contexts); 768baa89ba3SWambui Karuga drm_dbg(&i915->drm, "%s context support initialized\n", 769a4e7ccdaSChris Wilson DRIVER_CAPS(i915)->has_logical_contexts ? 77010be98a7SChris Wilson "logical" : "fake"); 77110be98a7SChris Wilson } 77210be98a7SChris Wilson 773a4e7ccdaSChris Wilson void i915_gem_driver_release__contexts(struct drm_i915_private *i915) 77410be98a7SChris Wilson { 7755f00cac9SChris Wilson flush_work(&i915->gem.contexts.free_work); 77610be98a7SChris Wilson } 77710be98a7SChris Wilson 77810be98a7SChris Wilson static int gem_context_register(struct i915_gem_context *ctx, 779c100777cSTvrtko Ursulin struct drm_i915_file_private *fpriv, 780c100777cSTvrtko Ursulin u32 *id) 78110be98a7SChris Wilson { 782a4e7ccdaSChris Wilson struct i915_address_space *vm; 78310be98a7SChris Wilson int ret; 78410be98a7SChris Wilson 78510be98a7SChris Wilson ctx->file_priv = fpriv; 786a4e7ccdaSChris Wilson 787a4e7ccdaSChris Wilson mutex_lock(&ctx->mutex); 788a4e7ccdaSChris Wilson vm = i915_gem_context_vm(ctx); 789a4e7ccdaSChris Wilson if (vm) 790a4e7ccdaSChris Wilson WRITE_ONCE(vm->file, fpriv); /* XXX */ 791a4e7ccdaSChris Wilson mutex_unlock(&ctx->mutex); 79210be98a7SChris Wilson 79310be98a7SChris Wilson ctx->pid = get_task_pid(current, PIDTYPE_PID); 794fc4f125dSChris Wilson snprintf(ctx->name, sizeof(ctx->name), "%s[%d]", 79510be98a7SChris Wilson current->comm, pid_nr(ctx->pid)); 79610be98a7SChris Wilson 79710be98a7SChris Wilson /* And finally expose ourselves to userspace via the idr */ 798c100777cSTvrtko Ursulin ret = xa_alloc(&fpriv->context_xa, id, ctx, xa_limit_32b, GFP_KERNEL); 799c100777cSTvrtko Ursulin if (ret) 80010be98a7SChris Wilson put_pid(fetch_and_zero(&ctx->pid)); 801c100777cSTvrtko Ursulin 80210be98a7SChris Wilson return ret; 80310be98a7SChris Wilson } 80410be98a7SChris Wilson 80510be98a7SChris Wilson int i915_gem_context_open(struct drm_i915_private *i915, 80610be98a7SChris Wilson struct drm_file *file) 80710be98a7SChris Wilson { 80810be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 80910be98a7SChris Wilson struct i915_gem_context *ctx; 81010be98a7SChris Wilson int err; 811c100777cSTvrtko Ursulin u32 id; 81210be98a7SChris Wilson 813c100777cSTvrtko Ursulin xa_init_flags(&file_priv->context_xa, XA_FLAGS_ALLOC); 814c100777cSTvrtko Ursulin 8155dbd2b7bSChris Wilson /* 0 reserved for invalid/unassigned ppgtt */ 8165dbd2b7bSChris Wilson xa_init_flags(&file_priv->vm_xa, XA_FLAGS_ALLOC1); 81710be98a7SChris Wilson 81810be98a7SChris Wilson ctx = i915_gem_create_context(i915, 0); 81910be98a7SChris Wilson if (IS_ERR(ctx)) { 82010be98a7SChris Wilson err = PTR_ERR(ctx); 82110be98a7SChris Wilson goto err; 82210be98a7SChris Wilson } 82310be98a7SChris Wilson 824c100777cSTvrtko Ursulin err = gem_context_register(ctx, file_priv, &id); 82510be98a7SChris Wilson if (err < 0) 82610be98a7SChris Wilson goto err_ctx; 82710be98a7SChris Wilson 828c100777cSTvrtko Ursulin GEM_BUG_ON(id); 82910be98a7SChris Wilson return 0; 83010be98a7SChris Wilson 83110be98a7SChris Wilson err_ctx: 83210be98a7SChris Wilson context_close(ctx); 83310be98a7SChris Wilson err: 8345dbd2b7bSChris Wilson xa_destroy(&file_priv->vm_xa); 835c100777cSTvrtko Ursulin xa_destroy(&file_priv->context_xa); 83610be98a7SChris Wilson return err; 83710be98a7SChris Wilson } 83810be98a7SChris Wilson 83910be98a7SChris Wilson void i915_gem_context_close(struct drm_file *file) 84010be98a7SChris Wilson { 84110be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 842a4e7ccdaSChris Wilson struct drm_i915_private *i915 = file_priv->dev_priv; 8435dbd2b7bSChris Wilson struct i915_address_space *vm; 844c100777cSTvrtko Ursulin struct i915_gem_context *ctx; 845c100777cSTvrtko Ursulin unsigned long idx; 84610be98a7SChris Wilson 847c100777cSTvrtko Ursulin xa_for_each(&file_priv->context_xa, idx, ctx) 848c100777cSTvrtko Ursulin context_close(ctx); 849c100777cSTvrtko Ursulin xa_destroy(&file_priv->context_xa); 85010be98a7SChris Wilson 8515dbd2b7bSChris Wilson xa_for_each(&file_priv->vm_xa, idx, vm) 8525dbd2b7bSChris Wilson i915_vm_put(vm); 8535dbd2b7bSChris Wilson xa_destroy(&file_priv->vm_xa); 854a4e7ccdaSChris Wilson 855a4e7ccdaSChris Wilson contexts_flush_free(&i915->gem.contexts); 85610be98a7SChris Wilson } 85710be98a7SChris Wilson 85810be98a7SChris Wilson int i915_gem_vm_create_ioctl(struct drm_device *dev, void *data, 85910be98a7SChris Wilson struct drm_file *file) 86010be98a7SChris Wilson { 86110be98a7SChris Wilson struct drm_i915_private *i915 = to_i915(dev); 86210be98a7SChris Wilson struct drm_i915_gem_vm_control *args = data; 86310be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 864ab53497bSChris Wilson struct i915_ppgtt *ppgtt; 8655dbd2b7bSChris Wilson u32 id; 86610be98a7SChris Wilson int err; 86710be98a7SChris Wilson 86810be98a7SChris Wilson if (!HAS_FULL_PPGTT(i915)) 86910be98a7SChris Wilson return -ENODEV; 87010be98a7SChris Wilson 87110be98a7SChris Wilson if (args->flags) 87210be98a7SChris Wilson return -EINVAL; 87310be98a7SChris Wilson 8742c86e55dSMatthew Auld ppgtt = i915_ppgtt_create(&i915->gt); 87510be98a7SChris Wilson if (IS_ERR(ppgtt)) 87610be98a7SChris Wilson return PTR_ERR(ppgtt); 87710be98a7SChris Wilson 87810be98a7SChris Wilson ppgtt->vm.file = file_priv; 87910be98a7SChris Wilson 88010be98a7SChris Wilson if (args->extensions) { 88110be98a7SChris Wilson err = i915_user_extensions(u64_to_user_ptr(args->extensions), 88210be98a7SChris Wilson NULL, 0, 88310be98a7SChris Wilson ppgtt); 88410be98a7SChris Wilson if (err) 88510be98a7SChris Wilson goto err_put; 88610be98a7SChris Wilson } 88710be98a7SChris Wilson 8885dbd2b7bSChris Wilson err = xa_alloc(&file_priv->vm_xa, &id, &ppgtt->vm, 8895dbd2b7bSChris Wilson xa_limit_32b, GFP_KERNEL); 89010be98a7SChris Wilson if (err) 89110be98a7SChris Wilson goto err_put; 89210be98a7SChris Wilson 8935dbd2b7bSChris Wilson GEM_BUG_ON(id == 0); /* reserved for invalid/unassigned ppgtt */ 8945dbd2b7bSChris Wilson args->vm_id = id; 89510be98a7SChris Wilson return 0; 89610be98a7SChris Wilson 89710be98a7SChris Wilson err_put: 898e568ac38SChris Wilson i915_vm_put(&ppgtt->vm); 89910be98a7SChris Wilson return err; 90010be98a7SChris Wilson } 90110be98a7SChris Wilson 90210be98a7SChris Wilson int i915_gem_vm_destroy_ioctl(struct drm_device *dev, void *data, 90310be98a7SChris Wilson struct drm_file *file) 90410be98a7SChris Wilson { 90510be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 90610be98a7SChris Wilson struct drm_i915_gem_vm_control *args = data; 907e568ac38SChris Wilson struct i915_address_space *vm; 90810be98a7SChris Wilson 90910be98a7SChris Wilson if (args->flags) 91010be98a7SChris Wilson return -EINVAL; 91110be98a7SChris Wilson 91210be98a7SChris Wilson if (args->extensions) 91310be98a7SChris Wilson return -EINVAL; 91410be98a7SChris Wilson 9155dbd2b7bSChris Wilson vm = xa_erase(&file_priv->vm_xa, args->vm_id); 916e568ac38SChris Wilson if (!vm) 91710be98a7SChris Wilson return -ENOENT; 91810be98a7SChris Wilson 919e568ac38SChris Wilson i915_vm_put(vm); 92010be98a7SChris Wilson return 0; 92110be98a7SChris Wilson } 92210be98a7SChris Wilson 92310be98a7SChris Wilson struct context_barrier_task { 92410be98a7SChris Wilson struct i915_active base; 92510be98a7SChris Wilson void (*task)(void *data); 92610be98a7SChris Wilson void *data; 92710be98a7SChris Wilson }; 92810be98a7SChris Wilson 929274cbf20SChris Wilson __i915_active_call 93010be98a7SChris Wilson static void cb_retire(struct i915_active *base) 93110be98a7SChris Wilson { 93210be98a7SChris Wilson struct context_barrier_task *cb = container_of(base, typeof(*cb), base); 93310be98a7SChris Wilson 93410be98a7SChris Wilson if (cb->task) 93510be98a7SChris Wilson cb->task(cb->data); 93610be98a7SChris Wilson 93710be98a7SChris Wilson i915_active_fini(&cb->base); 93810be98a7SChris Wilson kfree(cb); 93910be98a7SChris Wilson } 94010be98a7SChris Wilson 94110be98a7SChris Wilson I915_SELFTEST_DECLARE(static intel_engine_mask_t context_barrier_inject_fault); 94210be98a7SChris Wilson static int context_barrier_task(struct i915_gem_context *ctx, 94310be98a7SChris Wilson intel_engine_mask_t engines, 9441fe2d6f9SChris Wilson bool (*skip)(struct intel_context *ce, void *data), 94510be98a7SChris Wilson int (*emit)(struct i915_request *rq, void *data), 94610be98a7SChris Wilson void (*task)(void *data), 94710be98a7SChris Wilson void *data) 94810be98a7SChris Wilson { 94910be98a7SChris Wilson struct context_barrier_task *cb; 95010be98a7SChris Wilson struct i915_gem_engines_iter it; 95110be98a7SChris Wilson struct intel_context *ce; 95210be98a7SChris Wilson int err = 0; 95310be98a7SChris Wilson 95410be98a7SChris Wilson GEM_BUG_ON(!task); 95510be98a7SChris Wilson 95610be98a7SChris Wilson cb = kmalloc(sizeof(*cb), GFP_KERNEL); 95710be98a7SChris Wilson if (!cb) 95810be98a7SChris Wilson return -ENOMEM; 95910be98a7SChris Wilson 960b1e3177bSChris Wilson i915_active_init(&cb->base, NULL, cb_retire); 96112c255b5SChris Wilson err = i915_active_acquire(&cb->base); 96212c255b5SChris Wilson if (err) { 96312c255b5SChris Wilson kfree(cb); 96412c255b5SChris Wilson return err; 96512c255b5SChris Wilson } 96610be98a7SChris Wilson 96710be98a7SChris Wilson for_each_gem_engine(ce, i915_gem_context_lock_engines(ctx), it) { 96810be98a7SChris Wilson struct i915_request *rq; 96910be98a7SChris Wilson 97010be98a7SChris Wilson if (I915_SELFTEST_ONLY(context_barrier_inject_fault & 97110be98a7SChris Wilson ce->engine->mask)) { 97210be98a7SChris Wilson err = -ENXIO; 97310be98a7SChris Wilson break; 97410be98a7SChris Wilson } 97510be98a7SChris Wilson 9761fe2d6f9SChris Wilson if (!(ce->engine->mask & engines)) 9771fe2d6f9SChris Wilson continue; 9781fe2d6f9SChris Wilson 9791fe2d6f9SChris Wilson if (skip && skip(ce, data)) 98010be98a7SChris Wilson continue; 98110be98a7SChris Wilson 98210be98a7SChris Wilson rq = intel_context_create_request(ce); 98310be98a7SChris Wilson if (IS_ERR(rq)) { 98410be98a7SChris Wilson err = PTR_ERR(rq); 98510be98a7SChris Wilson break; 98610be98a7SChris Wilson } 98710be98a7SChris Wilson 98810be98a7SChris Wilson err = 0; 98910be98a7SChris Wilson if (emit) 99010be98a7SChris Wilson err = emit(rq, data); 99110be98a7SChris Wilson if (err == 0) 992d19d71fcSChris Wilson err = i915_active_add_request(&cb->base, rq); 99310be98a7SChris Wilson 99410be98a7SChris Wilson i915_request_add(rq); 99510be98a7SChris Wilson if (err) 99610be98a7SChris Wilson break; 99710be98a7SChris Wilson } 99810be98a7SChris Wilson i915_gem_context_unlock_engines(ctx); 99910be98a7SChris Wilson 100010be98a7SChris Wilson cb->task = err ? NULL : task; /* caller needs to unwind instead */ 100110be98a7SChris Wilson cb->data = data; 100210be98a7SChris Wilson 100310be98a7SChris Wilson i915_active_release(&cb->base); 100410be98a7SChris Wilson 100510be98a7SChris Wilson return err; 100610be98a7SChris Wilson } 100710be98a7SChris Wilson 100810be98a7SChris Wilson static int get_ppgtt(struct drm_i915_file_private *file_priv, 100910be98a7SChris Wilson struct i915_gem_context *ctx, 101010be98a7SChris Wilson struct drm_i915_gem_context_param *args) 101110be98a7SChris Wilson { 1012e568ac38SChris Wilson struct i915_address_space *vm; 10135dbd2b7bSChris Wilson int err; 10145dbd2b7bSChris Wilson u32 id; 101510be98a7SChris Wilson 1016a4e7ccdaSChris Wilson if (!rcu_access_pointer(ctx->vm)) 101710be98a7SChris Wilson return -ENODEV; 101810be98a7SChris Wilson 1019a4e7ccdaSChris Wilson rcu_read_lock(); 102027dbae8fSChris Wilson vm = context_get_vm_rcu(ctx); 1021a4e7ccdaSChris Wilson rcu_read_unlock(); 102290211ea4SChris Wilson if (!vm) 102390211ea4SChris Wilson return -ENODEV; 102490211ea4SChris Wilson 102590211ea4SChris Wilson err = xa_alloc(&file_priv->vm_xa, &id, vm, xa_limit_32b, GFP_KERNEL); 10265dbd2b7bSChris Wilson if (err) 102710be98a7SChris Wilson goto err_put; 102810be98a7SChris Wilson 10292850748eSChris Wilson i915_vm_open(vm); 103010be98a7SChris Wilson 10315dbd2b7bSChris Wilson GEM_BUG_ON(id == 0); /* reserved for invalid/unassigned ppgtt */ 10325dbd2b7bSChris Wilson args->value = id; 103310be98a7SChris Wilson args->size = 0; 103410be98a7SChris Wilson 103510be98a7SChris Wilson err_put: 1036e568ac38SChris Wilson i915_vm_put(vm); 10375dbd2b7bSChris Wilson return err; 103810be98a7SChris Wilson } 103910be98a7SChris Wilson 104010be98a7SChris Wilson static void set_ppgtt_barrier(void *data) 104110be98a7SChris Wilson { 1042e568ac38SChris Wilson struct i915_address_space *old = data; 104310be98a7SChris Wilson 1044e568ac38SChris Wilson if (INTEL_GEN(old->i915) < 8) 1045e568ac38SChris Wilson gen6_ppgtt_unpin_all(i915_vm_to_ppgtt(old)); 104610be98a7SChris Wilson 10472850748eSChris Wilson i915_vm_close(old); 104810be98a7SChris Wilson } 104910be98a7SChris Wilson 105010be98a7SChris Wilson static int emit_ppgtt_update(struct i915_request *rq, void *data) 105110be98a7SChris Wilson { 10529f3ccd40SChris Wilson struct i915_address_space *vm = rq->context->vm; 105310be98a7SChris Wilson struct intel_engine_cs *engine = rq->engine; 105410be98a7SChris Wilson u32 base = engine->mmio_base; 105510be98a7SChris Wilson u32 *cs; 105610be98a7SChris Wilson int i; 105710be98a7SChris Wilson 1058e568ac38SChris Wilson if (i915_vm_is_4lvl(vm)) { 1059ab53497bSChris Wilson struct i915_ppgtt *ppgtt = i915_vm_to_ppgtt(vm); 1060b5b7bef9SMika Kuoppala const dma_addr_t pd_daddr = px_dma(ppgtt->pd); 106110be98a7SChris Wilson 106210be98a7SChris Wilson cs = intel_ring_begin(rq, 6); 106310be98a7SChris Wilson if (IS_ERR(cs)) 106410be98a7SChris Wilson return PTR_ERR(cs); 106510be98a7SChris Wilson 106610be98a7SChris Wilson *cs++ = MI_LOAD_REGISTER_IMM(2); 106710be98a7SChris Wilson 106810be98a7SChris Wilson *cs++ = i915_mmio_reg_offset(GEN8_RING_PDP_UDW(base, 0)); 106910be98a7SChris Wilson *cs++ = upper_32_bits(pd_daddr); 107010be98a7SChris Wilson *cs++ = i915_mmio_reg_offset(GEN8_RING_PDP_LDW(base, 0)); 107110be98a7SChris Wilson *cs++ = lower_32_bits(pd_daddr); 107210be98a7SChris Wilson 107310be98a7SChris Wilson *cs++ = MI_NOOP; 107410be98a7SChris Wilson intel_ring_advance(rq, cs); 107510be98a7SChris Wilson } else if (HAS_LOGICAL_RING_CONTEXTS(engine->i915)) { 1076ab53497bSChris Wilson struct i915_ppgtt *ppgtt = i915_vm_to_ppgtt(vm); 1077191797a8SChris Wilson int err; 1078191797a8SChris Wilson 1079191797a8SChris Wilson /* Magic required to prevent forcewake errors! */ 1080191797a8SChris Wilson err = engine->emit_flush(rq, EMIT_INVALIDATE); 1081191797a8SChris Wilson if (err) 1082191797a8SChris Wilson return err; 1083e568ac38SChris Wilson 108410be98a7SChris Wilson cs = intel_ring_begin(rq, 4 * GEN8_3LVL_PDPES + 2); 108510be98a7SChris Wilson if (IS_ERR(cs)) 108610be98a7SChris Wilson return PTR_ERR(cs); 108710be98a7SChris Wilson 1088191797a8SChris Wilson *cs++ = MI_LOAD_REGISTER_IMM(2 * GEN8_3LVL_PDPES) | MI_LRI_FORCE_POSTED; 108910be98a7SChris Wilson for (i = GEN8_3LVL_PDPES; i--; ) { 109010be98a7SChris Wilson const dma_addr_t pd_daddr = i915_page_dir_dma_addr(ppgtt, i); 109110be98a7SChris Wilson 109210be98a7SChris Wilson *cs++ = i915_mmio_reg_offset(GEN8_RING_PDP_UDW(base, i)); 109310be98a7SChris Wilson *cs++ = upper_32_bits(pd_daddr); 109410be98a7SChris Wilson *cs++ = i915_mmio_reg_offset(GEN8_RING_PDP_LDW(base, i)); 109510be98a7SChris Wilson *cs++ = lower_32_bits(pd_daddr); 109610be98a7SChris Wilson } 109710be98a7SChris Wilson *cs++ = MI_NOOP; 109810be98a7SChris Wilson intel_ring_advance(rq, cs); 109910be98a7SChris Wilson } 110010be98a7SChris Wilson 110110be98a7SChris Wilson return 0; 110210be98a7SChris Wilson } 110310be98a7SChris Wilson 11041fe2d6f9SChris Wilson static bool skip_ppgtt_update(struct intel_context *ce, void *data) 11051fe2d6f9SChris Wilson { 1106aef82079SChris Wilson if (!test_bit(CONTEXT_ALLOC_BIT, &ce->flags)) 1107aef82079SChris Wilson return true; 1108aef82079SChris Wilson 11091fe2d6f9SChris Wilson if (HAS_LOGICAL_RING_CONTEXTS(ce->engine->i915)) 1110aef82079SChris Wilson return false; 1111aef82079SChris Wilson 1112aef82079SChris Wilson if (!atomic_read(&ce->pin_count)) 1113aef82079SChris Wilson return true; 1114aef82079SChris Wilson 1115aef82079SChris Wilson /* ppGTT is not part of the legacy context image */ 1116aef82079SChris Wilson if (gen6_ppgtt_pin(i915_vm_to_ppgtt(ce->vm))) 1117aef82079SChris Wilson return true; 1118aef82079SChris Wilson 1119aef82079SChris Wilson return false; 11201fe2d6f9SChris Wilson } 11211fe2d6f9SChris Wilson 112210be98a7SChris Wilson static int set_ppgtt(struct drm_i915_file_private *file_priv, 112310be98a7SChris Wilson struct i915_gem_context *ctx, 112410be98a7SChris Wilson struct drm_i915_gem_context_param *args) 112510be98a7SChris Wilson { 1126e568ac38SChris Wilson struct i915_address_space *vm, *old; 112710be98a7SChris Wilson int err; 112810be98a7SChris Wilson 112910be98a7SChris Wilson if (args->size) 113010be98a7SChris Wilson return -EINVAL; 113110be98a7SChris Wilson 1132a4e7ccdaSChris Wilson if (!rcu_access_pointer(ctx->vm)) 113310be98a7SChris Wilson return -ENODEV; 113410be98a7SChris Wilson 113510be98a7SChris Wilson if (upper_32_bits(args->value)) 113610be98a7SChris Wilson return -ENOENT; 113710be98a7SChris Wilson 1138aabbe344SChris Wilson rcu_read_lock(); 11395dbd2b7bSChris Wilson vm = xa_load(&file_priv->vm_xa, args->value); 1140aabbe344SChris Wilson if (vm && !kref_get_unless_zero(&vm->ref)) 1141aabbe344SChris Wilson vm = NULL; 1142aabbe344SChris Wilson rcu_read_unlock(); 1143e568ac38SChris Wilson if (!vm) 114410be98a7SChris Wilson return -ENOENT; 114510be98a7SChris Wilson 1146a4e7ccdaSChris Wilson err = mutex_lock_interruptible(&ctx->mutex); 114710be98a7SChris Wilson if (err) 114810be98a7SChris Wilson goto out; 114910be98a7SChris Wilson 1150a4e7ccdaSChris Wilson if (i915_gem_context_is_closed(ctx)) { 1151a4e7ccdaSChris Wilson err = -ENOENT; 1152feba2b81SChris Wilson goto unlock; 1153a4e7ccdaSChris Wilson } 1154a4e7ccdaSChris Wilson 1155a4e7ccdaSChris Wilson if (vm == rcu_access_pointer(ctx->vm)) 115610be98a7SChris Wilson goto unlock; 115710be98a7SChris Wilson 115810be98a7SChris Wilson /* Teardown the existing obj:vma cache, it will have to be rebuilt. */ 115910be98a7SChris Wilson lut_close(ctx); 116010be98a7SChris Wilson 1161e568ac38SChris Wilson old = __set_ppgtt(ctx, vm); 116210be98a7SChris Wilson 116310be98a7SChris Wilson /* 116410be98a7SChris Wilson * We need to flush any requests using the current ppgtt before 116510be98a7SChris Wilson * we release it as the requests do not hold a reference themselves, 116610be98a7SChris Wilson * only indirectly through the context. 116710be98a7SChris Wilson */ 116810be98a7SChris Wilson err = context_barrier_task(ctx, ALL_ENGINES, 11691fe2d6f9SChris Wilson skip_ppgtt_update, 117010be98a7SChris Wilson emit_ppgtt_update, 117110be98a7SChris Wilson set_ppgtt_barrier, 117210be98a7SChris Wilson old); 117310be98a7SChris Wilson if (err) { 11742850748eSChris Wilson i915_vm_close(__set_ppgtt(ctx, old)); 11752850748eSChris Wilson i915_vm_close(old); 117610be98a7SChris Wilson } 117710be98a7SChris Wilson 117810be98a7SChris Wilson unlock: 1179a4e7ccdaSChris Wilson mutex_unlock(&ctx->mutex); 118010be98a7SChris Wilson out: 1181e568ac38SChris Wilson i915_vm_put(vm); 118210be98a7SChris Wilson return err; 118310be98a7SChris Wilson } 118410be98a7SChris Wilson 118510be98a7SChris Wilson static int 118610be98a7SChris Wilson user_to_context_sseu(struct drm_i915_private *i915, 118710be98a7SChris Wilson const struct drm_i915_gem_context_param_sseu *user, 118810be98a7SChris Wilson struct intel_sseu *context) 118910be98a7SChris Wilson { 119010be98a7SChris Wilson const struct sseu_dev_info *device = &RUNTIME_INFO(i915)->sseu; 119110be98a7SChris Wilson 119210be98a7SChris Wilson /* No zeros in any field. */ 119310be98a7SChris Wilson if (!user->slice_mask || !user->subslice_mask || 119410be98a7SChris Wilson !user->min_eus_per_subslice || !user->max_eus_per_subslice) 119510be98a7SChris Wilson return -EINVAL; 119610be98a7SChris Wilson 119710be98a7SChris Wilson /* Max > min. */ 119810be98a7SChris Wilson if (user->max_eus_per_subslice < user->min_eus_per_subslice) 119910be98a7SChris Wilson return -EINVAL; 120010be98a7SChris Wilson 120110be98a7SChris Wilson /* 120210be98a7SChris Wilson * Some future proofing on the types since the uAPI is wider than the 120310be98a7SChris Wilson * current internal implementation. 120410be98a7SChris Wilson */ 120510be98a7SChris Wilson if (overflows_type(user->slice_mask, context->slice_mask) || 120610be98a7SChris Wilson overflows_type(user->subslice_mask, context->subslice_mask) || 120710be98a7SChris Wilson overflows_type(user->min_eus_per_subslice, 120810be98a7SChris Wilson context->min_eus_per_subslice) || 120910be98a7SChris Wilson overflows_type(user->max_eus_per_subslice, 121010be98a7SChris Wilson context->max_eus_per_subslice)) 121110be98a7SChris Wilson return -EINVAL; 121210be98a7SChris Wilson 121310be98a7SChris Wilson /* Check validity against hardware. */ 121410be98a7SChris Wilson if (user->slice_mask & ~device->slice_mask) 121510be98a7SChris Wilson return -EINVAL; 121610be98a7SChris Wilson 121710be98a7SChris Wilson if (user->subslice_mask & ~device->subslice_mask[0]) 121810be98a7SChris Wilson return -EINVAL; 121910be98a7SChris Wilson 122010be98a7SChris Wilson if (user->max_eus_per_subslice > device->max_eus_per_subslice) 122110be98a7SChris Wilson return -EINVAL; 122210be98a7SChris Wilson 122310be98a7SChris Wilson context->slice_mask = user->slice_mask; 122410be98a7SChris Wilson context->subslice_mask = user->subslice_mask; 122510be98a7SChris Wilson context->min_eus_per_subslice = user->min_eus_per_subslice; 122610be98a7SChris Wilson context->max_eus_per_subslice = user->max_eus_per_subslice; 122710be98a7SChris Wilson 122810be98a7SChris Wilson /* Part specific restrictions. */ 122910be98a7SChris Wilson if (IS_GEN(i915, 11)) { 123010be98a7SChris Wilson unsigned int hw_s = hweight8(device->slice_mask); 123110be98a7SChris Wilson unsigned int hw_ss_per_s = hweight8(device->subslice_mask[0]); 123210be98a7SChris Wilson unsigned int req_s = hweight8(context->slice_mask); 123310be98a7SChris Wilson unsigned int req_ss = hweight8(context->subslice_mask); 123410be98a7SChris Wilson 123510be98a7SChris Wilson /* 123610be98a7SChris Wilson * Only full subslice enablement is possible if more than one 123710be98a7SChris Wilson * slice is turned on. 123810be98a7SChris Wilson */ 123910be98a7SChris Wilson if (req_s > 1 && req_ss != hw_ss_per_s) 124010be98a7SChris Wilson return -EINVAL; 124110be98a7SChris Wilson 124210be98a7SChris Wilson /* 124310be98a7SChris Wilson * If more than four (SScount bitfield limit) subslices are 124410be98a7SChris Wilson * requested then the number has to be even. 124510be98a7SChris Wilson */ 124610be98a7SChris Wilson if (req_ss > 4 && (req_ss & 1)) 124710be98a7SChris Wilson return -EINVAL; 124810be98a7SChris Wilson 124910be98a7SChris Wilson /* 125010be98a7SChris Wilson * If only one slice is enabled and subslice count is below the 125110be98a7SChris Wilson * device full enablement, it must be at most half of the all 125210be98a7SChris Wilson * available subslices. 125310be98a7SChris Wilson */ 125410be98a7SChris Wilson if (req_s == 1 && req_ss < hw_ss_per_s && 125510be98a7SChris Wilson req_ss > (hw_ss_per_s / 2)) 125610be98a7SChris Wilson return -EINVAL; 125710be98a7SChris Wilson 125810be98a7SChris Wilson /* ABI restriction - VME use case only. */ 125910be98a7SChris Wilson 126010be98a7SChris Wilson /* All slices or one slice only. */ 126110be98a7SChris Wilson if (req_s != 1 && req_s != hw_s) 126210be98a7SChris Wilson return -EINVAL; 126310be98a7SChris Wilson 126410be98a7SChris Wilson /* 126510be98a7SChris Wilson * Half subslices or full enablement only when one slice is 126610be98a7SChris Wilson * enabled. 126710be98a7SChris Wilson */ 126810be98a7SChris Wilson if (req_s == 1 && 126910be98a7SChris Wilson (req_ss != hw_ss_per_s && req_ss != (hw_ss_per_s / 2))) 127010be98a7SChris Wilson return -EINVAL; 127110be98a7SChris Wilson 127210be98a7SChris Wilson /* No EU configuration changes. */ 127310be98a7SChris Wilson if ((user->min_eus_per_subslice != 127410be98a7SChris Wilson device->max_eus_per_subslice) || 127510be98a7SChris Wilson (user->max_eus_per_subslice != 127610be98a7SChris Wilson device->max_eus_per_subslice)) 127710be98a7SChris Wilson return -EINVAL; 127810be98a7SChris Wilson } 127910be98a7SChris Wilson 128010be98a7SChris Wilson return 0; 128110be98a7SChris Wilson } 128210be98a7SChris Wilson 128310be98a7SChris Wilson static int set_sseu(struct i915_gem_context *ctx, 128410be98a7SChris Wilson struct drm_i915_gem_context_param *args) 128510be98a7SChris Wilson { 128610be98a7SChris Wilson struct drm_i915_private *i915 = ctx->i915; 128710be98a7SChris Wilson struct drm_i915_gem_context_param_sseu user_sseu; 128810be98a7SChris Wilson struct intel_context *ce; 128910be98a7SChris Wilson struct intel_sseu sseu; 129010be98a7SChris Wilson unsigned long lookup; 129110be98a7SChris Wilson int ret; 129210be98a7SChris Wilson 129310be98a7SChris Wilson if (args->size < sizeof(user_sseu)) 129410be98a7SChris Wilson return -EINVAL; 129510be98a7SChris Wilson 129610be98a7SChris Wilson if (!IS_GEN(i915, 11)) 129710be98a7SChris Wilson return -ENODEV; 129810be98a7SChris Wilson 129910be98a7SChris Wilson if (copy_from_user(&user_sseu, u64_to_user_ptr(args->value), 130010be98a7SChris Wilson sizeof(user_sseu))) 130110be98a7SChris Wilson return -EFAULT; 130210be98a7SChris Wilson 130310be98a7SChris Wilson if (user_sseu.rsvd) 130410be98a7SChris Wilson return -EINVAL; 130510be98a7SChris Wilson 130610be98a7SChris Wilson if (user_sseu.flags & ~(I915_CONTEXT_SSEU_FLAG_ENGINE_INDEX)) 130710be98a7SChris Wilson return -EINVAL; 130810be98a7SChris Wilson 130910be98a7SChris Wilson lookup = 0; 131010be98a7SChris Wilson if (user_sseu.flags & I915_CONTEXT_SSEU_FLAG_ENGINE_INDEX) 131110be98a7SChris Wilson lookup |= LOOKUP_USER_INDEX; 131210be98a7SChris Wilson 131310be98a7SChris Wilson ce = lookup_user_engine(ctx, lookup, &user_sseu.engine); 131410be98a7SChris Wilson if (IS_ERR(ce)) 131510be98a7SChris Wilson return PTR_ERR(ce); 131610be98a7SChris Wilson 131710be98a7SChris Wilson /* Only render engine supports RPCS configuration. */ 131810be98a7SChris Wilson if (ce->engine->class != RENDER_CLASS) { 131910be98a7SChris Wilson ret = -ENODEV; 132010be98a7SChris Wilson goto out_ce; 132110be98a7SChris Wilson } 132210be98a7SChris Wilson 132310be98a7SChris Wilson ret = user_to_context_sseu(i915, &user_sseu, &sseu); 132410be98a7SChris Wilson if (ret) 132510be98a7SChris Wilson goto out_ce; 132610be98a7SChris Wilson 132710be98a7SChris Wilson ret = intel_context_reconfigure_sseu(ce, sseu); 132810be98a7SChris Wilson if (ret) 132910be98a7SChris Wilson goto out_ce; 133010be98a7SChris Wilson 133110be98a7SChris Wilson args->size = sizeof(user_sseu); 133210be98a7SChris Wilson 133310be98a7SChris Wilson out_ce: 133410be98a7SChris Wilson intel_context_put(ce); 133510be98a7SChris Wilson return ret; 133610be98a7SChris Wilson } 133710be98a7SChris Wilson 133810be98a7SChris Wilson struct set_engines { 133910be98a7SChris Wilson struct i915_gem_context *ctx; 134010be98a7SChris Wilson struct i915_gem_engines *engines; 134110be98a7SChris Wilson }; 134210be98a7SChris Wilson 134310be98a7SChris Wilson static int 134410be98a7SChris Wilson set_engines__load_balance(struct i915_user_extension __user *base, void *data) 134510be98a7SChris Wilson { 134610be98a7SChris Wilson struct i915_context_engines_load_balance __user *ext = 134710be98a7SChris Wilson container_of_user(base, typeof(*ext), base); 134810be98a7SChris Wilson const struct set_engines *set = data; 1349d0bf4582SWambui Karuga struct drm_i915_private *i915 = set->ctx->i915; 135010be98a7SChris Wilson struct intel_engine_cs *stack[16]; 135110be98a7SChris Wilson struct intel_engine_cs **siblings; 135210be98a7SChris Wilson struct intel_context *ce; 135310be98a7SChris Wilson u16 num_siblings, idx; 135410be98a7SChris Wilson unsigned int n; 135510be98a7SChris Wilson int err; 135610be98a7SChris Wilson 1357d0bf4582SWambui Karuga if (!HAS_EXECLISTS(i915)) 135810be98a7SChris Wilson return -ENODEV; 135910be98a7SChris Wilson 1360d0bf4582SWambui Karuga if (USES_GUC_SUBMISSION(i915)) 136110be98a7SChris Wilson return -ENODEV; /* not implement yet */ 136210be98a7SChris Wilson 136310be98a7SChris Wilson if (get_user(idx, &ext->engine_index)) 136410be98a7SChris Wilson return -EFAULT; 136510be98a7SChris Wilson 136610be98a7SChris Wilson if (idx >= set->engines->num_engines) { 1367d0bf4582SWambui Karuga drm_dbg(&i915->drm, "Invalid placement value, %d >= %d\n", 136810be98a7SChris Wilson idx, set->engines->num_engines); 136910be98a7SChris Wilson return -EINVAL; 137010be98a7SChris Wilson } 137110be98a7SChris Wilson 137210be98a7SChris Wilson idx = array_index_nospec(idx, set->engines->num_engines); 137310be98a7SChris Wilson if (set->engines->engines[idx]) { 1374d0bf4582SWambui Karuga drm_dbg(&i915->drm, 1375d0bf4582SWambui Karuga "Invalid placement[%d], already occupied\n", idx); 137610be98a7SChris Wilson return -EEXIST; 137710be98a7SChris Wilson } 137810be98a7SChris Wilson 137910be98a7SChris Wilson if (get_user(num_siblings, &ext->num_siblings)) 138010be98a7SChris Wilson return -EFAULT; 138110be98a7SChris Wilson 138210be98a7SChris Wilson err = check_user_mbz(&ext->flags); 138310be98a7SChris Wilson if (err) 138410be98a7SChris Wilson return err; 138510be98a7SChris Wilson 138610be98a7SChris Wilson err = check_user_mbz(&ext->mbz64); 138710be98a7SChris Wilson if (err) 138810be98a7SChris Wilson return err; 138910be98a7SChris Wilson 139010be98a7SChris Wilson siblings = stack; 139110be98a7SChris Wilson if (num_siblings > ARRAY_SIZE(stack)) { 139210be98a7SChris Wilson siblings = kmalloc_array(num_siblings, 139310be98a7SChris Wilson sizeof(*siblings), 139410be98a7SChris Wilson GFP_KERNEL); 139510be98a7SChris Wilson if (!siblings) 139610be98a7SChris Wilson return -ENOMEM; 139710be98a7SChris Wilson } 139810be98a7SChris Wilson 139910be98a7SChris Wilson for (n = 0; n < num_siblings; n++) { 140010be98a7SChris Wilson struct i915_engine_class_instance ci; 140110be98a7SChris Wilson 140210be98a7SChris Wilson if (copy_from_user(&ci, &ext->engines[n], sizeof(ci))) { 140310be98a7SChris Wilson err = -EFAULT; 140410be98a7SChris Wilson goto out_siblings; 140510be98a7SChris Wilson } 140610be98a7SChris Wilson 1407d0bf4582SWambui Karuga siblings[n] = intel_engine_lookup_user(i915, 140810be98a7SChris Wilson ci.engine_class, 140910be98a7SChris Wilson ci.engine_instance); 141010be98a7SChris Wilson if (!siblings[n]) { 1411d0bf4582SWambui Karuga drm_dbg(&i915->drm, 1412d0bf4582SWambui Karuga "Invalid sibling[%d]: { class:%d, inst:%d }\n", 141310be98a7SChris Wilson n, ci.engine_class, ci.engine_instance); 141410be98a7SChris Wilson err = -EINVAL; 141510be98a7SChris Wilson goto out_siblings; 141610be98a7SChris Wilson } 141710be98a7SChris Wilson } 141810be98a7SChris Wilson 1419e6ba7648SChris Wilson ce = intel_execlists_create_virtual(siblings, n); 142010be98a7SChris Wilson if (IS_ERR(ce)) { 142110be98a7SChris Wilson err = PTR_ERR(ce); 142210be98a7SChris Wilson goto out_siblings; 142310be98a7SChris Wilson } 142410be98a7SChris Wilson 1425e6ba7648SChris Wilson intel_context_set_gem(ce, set->ctx); 1426e6ba7648SChris Wilson 142710be98a7SChris Wilson if (cmpxchg(&set->engines->engines[idx], NULL, ce)) { 142810be98a7SChris Wilson intel_context_put(ce); 142910be98a7SChris Wilson err = -EEXIST; 143010be98a7SChris Wilson goto out_siblings; 143110be98a7SChris Wilson } 143210be98a7SChris Wilson 143310be98a7SChris Wilson out_siblings: 143410be98a7SChris Wilson if (siblings != stack) 143510be98a7SChris Wilson kfree(siblings); 143610be98a7SChris Wilson 143710be98a7SChris Wilson return err; 143810be98a7SChris Wilson } 143910be98a7SChris Wilson 144010be98a7SChris Wilson static int 144110be98a7SChris Wilson set_engines__bond(struct i915_user_extension __user *base, void *data) 144210be98a7SChris Wilson { 144310be98a7SChris Wilson struct i915_context_engines_bond __user *ext = 144410be98a7SChris Wilson container_of_user(base, typeof(*ext), base); 144510be98a7SChris Wilson const struct set_engines *set = data; 1446d0bf4582SWambui Karuga struct drm_i915_private *i915 = set->ctx->i915; 144710be98a7SChris Wilson struct i915_engine_class_instance ci; 144810be98a7SChris Wilson struct intel_engine_cs *virtual; 144910be98a7SChris Wilson struct intel_engine_cs *master; 145010be98a7SChris Wilson u16 idx, num_bonds; 145110be98a7SChris Wilson int err, n; 145210be98a7SChris Wilson 145310be98a7SChris Wilson if (get_user(idx, &ext->virtual_index)) 145410be98a7SChris Wilson return -EFAULT; 145510be98a7SChris Wilson 145610be98a7SChris Wilson if (idx >= set->engines->num_engines) { 1457d0bf4582SWambui Karuga drm_dbg(&i915->drm, 1458d0bf4582SWambui Karuga "Invalid index for virtual engine: %d >= %d\n", 145910be98a7SChris Wilson idx, set->engines->num_engines); 146010be98a7SChris Wilson return -EINVAL; 146110be98a7SChris Wilson } 146210be98a7SChris Wilson 146310be98a7SChris Wilson idx = array_index_nospec(idx, set->engines->num_engines); 146410be98a7SChris Wilson if (!set->engines->engines[idx]) { 1465d0bf4582SWambui Karuga drm_dbg(&i915->drm, "Invalid engine at %d\n", idx); 146610be98a7SChris Wilson return -EINVAL; 146710be98a7SChris Wilson } 146810be98a7SChris Wilson virtual = set->engines->engines[idx]->engine; 146910be98a7SChris Wilson 147010be98a7SChris Wilson err = check_user_mbz(&ext->flags); 147110be98a7SChris Wilson if (err) 147210be98a7SChris Wilson return err; 147310be98a7SChris Wilson 147410be98a7SChris Wilson for (n = 0; n < ARRAY_SIZE(ext->mbz64); n++) { 147510be98a7SChris Wilson err = check_user_mbz(&ext->mbz64[n]); 147610be98a7SChris Wilson if (err) 147710be98a7SChris Wilson return err; 147810be98a7SChris Wilson } 147910be98a7SChris Wilson 148010be98a7SChris Wilson if (copy_from_user(&ci, &ext->master, sizeof(ci))) 148110be98a7SChris Wilson return -EFAULT; 148210be98a7SChris Wilson 1483d0bf4582SWambui Karuga master = intel_engine_lookup_user(i915, 148410be98a7SChris Wilson ci.engine_class, ci.engine_instance); 148510be98a7SChris Wilson if (!master) { 1486d0bf4582SWambui Karuga drm_dbg(&i915->drm, 1487d0bf4582SWambui Karuga "Unrecognised master engine: { class:%u, instance:%u }\n", 148810be98a7SChris Wilson ci.engine_class, ci.engine_instance); 148910be98a7SChris Wilson return -EINVAL; 149010be98a7SChris Wilson } 149110be98a7SChris Wilson 149210be98a7SChris Wilson if (get_user(num_bonds, &ext->num_bonds)) 149310be98a7SChris Wilson return -EFAULT; 149410be98a7SChris Wilson 149510be98a7SChris Wilson for (n = 0; n < num_bonds; n++) { 149610be98a7SChris Wilson struct intel_engine_cs *bond; 149710be98a7SChris Wilson 149810be98a7SChris Wilson if (copy_from_user(&ci, &ext->engines[n], sizeof(ci))) 149910be98a7SChris Wilson return -EFAULT; 150010be98a7SChris Wilson 1501d0bf4582SWambui Karuga bond = intel_engine_lookup_user(i915, 150210be98a7SChris Wilson ci.engine_class, 150310be98a7SChris Wilson ci.engine_instance); 150410be98a7SChris Wilson if (!bond) { 1505d0bf4582SWambui Karuga drm_dbg(&i915->drm, 1506d0bf4582SWambui Karuga "Unrecognised engine[%d] for bonding: { class:%d, instance: %d }\n", 150710be98a7SChris Wilson n, ci.engine_class, ci.engine_instance); 150810be98a7SChris Wilson return -EINVAL; 150910be98a7SChris Wilson } 151010be98a7SChris Wilson 151110be98a7SChris Wilson /* 151210be98a7SChris Wilson * A non-virtual engine has no siblings to choose between; and 151310be98a7SChris Wilson * a submit fence will always be directed to the one engine. 151410be98a7SChris Wilson */ 151510be98a7SChris Wilson if (intel_engine_is_virtual(virtual)) { 151610be98a7SChris Wilson err = intel_virtual_engine_attach_bond(virtual, 151710be98a7SChris Wilson master, 151810be98a7SChris Wilson bond); 151910be98a7SChris Wilson if (err) 152010be98a7SChris Wilson return err; 152110be98a7SChris Wilson } 152210be98a7SChris Wilson } 152310be98a7SChris Wilson 152410be98a7SChris Wilson return 0; 152510be98a7SChris Wilson } 152610be98a7SChris Wilson 152710be98a7SChris Wilson static const i915_user_extension_fn set_engines__extensions[] = { 152810be98a7SChris Wilson [I915_CONTEXT_ENGINES_EXT_LOAD_BALANCE] = set_engines__load_balance, 152910be98a7SChris Wilson [I915_CONTEXT_ENGINES_EXT_BOND] = set_engines__bond, 153010be98a7SChris Wilson }; 153110be98a7SChris Wilson 153210be98a7SChris Wilson static int 153310be98a7SChris Wilson set_engines(struct i915_gem_context *ctx, 153410be98a7SChris Wilson const struct drm_i915_gem_context_param *args) 153510be98a7SChris Wilson { 1536baa89ba3SWambui Karuga struct drm_i915_private *i915 = ctx->i915; 153710be98a7SChris Wilson struct i915_context_param_engines __user *user = 153810be98a7SChris Wilson u64_to_user_ptr(args->value); 153910be98a7SChris Wilson struct set_engines set = { .ctx = ctx }; 154010be98a7SChris Wilson unsigned int num_engines, n; 154110be98a7SChris Wilson u64 extensions; 154210be98a7SChris Wilson int err; 154310be98a7SChris Wilson 154410be98a7SChris Wilson if (!args->size) { /* switch back to legacy user_ring_map */ 154510be98a7SChris Wilson if (!i915_gem_context_user_engines(ctx)) 154610be98a7SChris Wilson return 0; 154710be98a7SChris Wilson 154810be98a7SChris Wilson set.engines = default_engines(ctx); 154910be98a7SChris Wilson if (IS_ERR(set.engines)) 155010be98a7SChris Wilson return PTR_ERR(set.engines); 155110be98a7SChris Wilson 155210be98a7SChris Wilson goto replace; 155310be98a7SChris Wilson } 155410be98a7SChris Wilson 155510be98a7SChris Wilson BUILD_BUG_ON(!IS_ALIGNED(sizeof(*user), sizeof(*user->engines))); 155610be98a7SChris Wilson if (args->size < sizeof(*user) || 155710be98a7SChris Wilson !IS_ALIGNED(args->size, sizeof(*user->engines))) { 1558baa89ba3SWambui Karuga drm_dbg(&i915->drm, "Invalid size for engine array: %d\n", 155910be98a7SChris Wilson args->size); 156010be98a7SChris Wilson return -EINVAL; 156110be98a7SChris Wilson } 156210be98a7SChris Wilson 156310be98a7SChris Wilson /* 156410be98a7SChris Wilson * Note that I915_EXEC_RING_MASK limits execbuf to only using the 156510be98a7SChris Wilson * first 64 engines defined here. 156610be98a7SChris Wilson */ 156710be98a7SChris Wilson num_engines = (args->size - sizeof(*user)) / sizeof(*user->engines); 156810be98a7SChris Wilson 156910be98a7SChris Wilson set.engines = kmalloc(struct_size(set.engines, engines, num_engines), 157010be98a7SChris Wilson GFP_KERNEL); 157110be98a7SChris Wilson if (!set.engines) 157210be98a7SChris Wilson return -ENOMEM; 157310be98a7SChris Wilson 1574155ab883SChris Wilson init_rcu_head(&set.engines->rcu); 157510be98a7SChris Wilson for (n = 0; n < num_engines; n++) { 157610be98a7SChris Wilson struct i915_engine_class_instance ci; 157710be98a7SChris Wilson struct intel_engine_cs *engine; 157848ae397bSChris Wilson struct intel_context *ce; 157910be98a7SChris Wilson 158010be98a7SChris Wilson if (copy_from_user(&ci, &user->engines[n], sizeof(ci))) { 158110be98a7SChris Wilson __free_engines(set.engines, n); 158210be98a7SChris Wilson return -EFAULT; 158310be98a7SChris Wilson } 158410be98a7SChris Wilson 158510be98a7SChris Wilson if (ci.engine_class == (u16)I915_ENGINE_CLASS_INVALID && 158610be98a7SChris Wilson ci.engine_instance == (u16)I915_ENGINE_CLASS_INVALID_NONE) { 158710be98a7SChris Wilson set.engines->engines[n] = NULL; 158810be98a7SChris Wilson continue; 158910be98a7SChris Wilson } 159010be98a7SChris Wilson 159110be98a7SChris Wilson engine = intel_engine_lookup_user(ctx->i915, 159210be98a7SChris Wilson ci.engine_class, 159310be98a7SChris Wilson ci.engine_instance); 159410be98a7SChris Wilson if (!engine) { 1595baa89ba3SWambui Karuga drm_dbg(&i915->drm, 1596baa89ba3SWambui Karuga "Invalid engine[%d]: { class:%d, instance:%d }\n", 159710be98a7SChris Wilson n, ci.engine_class, ci.engine_instance); 159810be98a7SChris Wilson __free_engines(set.engines, n); 159910be98a7SChris Wilson return -ENOENT; 160010be98a7SChris Wilson } 160110be98a7SChris Wilson 1602e6ba7648SChris Wilson ce = intel_context_create(engine); 160348ae397bSChris Wilson if (IS_ERR(ce)) { 160410be98a7SChris Wilson __free_engines(set.engines, n); 160548ae397bSChris Wilson return PTR_ERR(ce); 160610be98a7SChris Wilson } 160748ae397bSChris Wilson 1608e6ba7648SChris Wilson intel_context_set_gem(ce, ctx); 1609e6ba7648SChris Wilson 161048ae397bSChris Wilson set.engines->engines[n] = ce; 161110be98a7SChris Wilson } 161210be98a7SChris Wilson set.engines->num_engines = num_engines; 161310be98a7SChris Wilson 161410be98a7SChris Wilson err = -EFAULT; 161510be98a7SChris Wilson if (!get_user(extensions, &user->extensions)) 161610be98a7SChris Wilson err = i915_user_extensions(u64_to_user_ptr(extensions), 161710be98a7SChris Wilson set_engines__extensions, 161810be98a7SChris Wilson ARRAY_SIZE(set_engines__extensions), 161910be98a7SChris Wilson &set); 162010be98a7SChris Wilson if (err) { 162110be98a7SChris Wilson free_engines(set.engines); 162210be98a7SChris Wilson return err; 162310be98a7SChris Wilson } 162410be98a7SChris Wilson 162510be98a7SChris Wilson replace: 162610be98a7SChris Wilson mutex_lock(&ctx->engines_mutex); 162710be98a7SChris Wilson if (args->size) 162810be98a7SChris Wilson i915_gem_context_set_user_engines(ctx); 162910be98a7SChris Wilson else 163010be98a7SChris Wilson i915_gem_context_clear_user_engines(ctx); 16311feace5dSPaul E. McKenney set.engines = rcu_replace_pointer(ctx->engines, set.engines, 1); 163210be98a7SChris Wilson mutex_unlock(&ctx->engines_mutex); 163310be98a7SChris Wilson 1634155ab883SChris Wilson call_rcu(&set.engines->rcu, free_engines_rcu); 163510be98a7SChris Wilson 163610be98a7SChris Wilson return 0; 163710be98a7SChris Wilson } 163810be98a7SChris Wilson 163910be98a7SChris Wilson static struct i915_gem_engines * 164010be98a7SChris Wilson __copy_engines(struct i915_gem_engines *e) 164110be98a7SChris Wilson { 164210be98a7SChris Wilson struct i915_gem_engines *copy; 164310be98a7SChris Wilson unsigned int n; 164410be98a7SChris Wilson 164510be98a7SChris Wilson copy = kmalloc(struct_size(e, engines, e->num_engines), GFP_KERNEL); 164610be98a7SChris Wilson if (!copy) 164710be98a7SChris Wilson return ERR_PTR(-ENOMEM); 164810be98a7SChris Wilson 1649155ab883SChris Wilson init_rcu_head(©->rcu); 165010be98a7SChris Wilson for (n = 0; n < e->num_engines; n++) { 165110be98a7SChris Wilson if (e->engines[n]) 165210be98a7SChris Wilson copy->engines[n] = intel_context_get(e->engines[n]); 165310be98a7SChris Wilson else 165410be98a7SChris Wilson copy->engines[n] = NULL; 165510be98a7SChris Wilson } 165610be98a7SChris Wilson copy->num_engines = n; 165710be98a7SChris Wilson 165810be98a7SChris Wilson return copy; 165910be98a7SChris Wilson } 166010be98a7SChris Wilson 166110be98a7SChris Wilson static int 166210be98a7SChris Wilson get_engines(struct i915_gem_context *ctx, 166310be98a7SChris Wilson struct drm_i915_gem_context_param *args) 166410be98a7SChris Wilson { 166510be98a7SChris Wilson struct i915_context_param_engines __user *user; 166610be98a7SChris Wilson struct i915_gem_engines *e; 166710be98a7SChris Wilson size_t n, count, size; 166810be98a7SChris Wilson int err = 0; 166910be98a7SChris Wilson 167010be98a7SChris Wilson err = mutex_lock_interruptible(&ctx->engines_mutex); 167110be98a7SChris Wilson if (err) 167210be98a7SChris Wilson return err; 167310be98a7SChris Wilson 167410be98a7SChris Wilson e = NULL; 167510be98a7SChris Wilson if (i915_gem_context_user_engines(ctx)) 167610be98a7SChris Wilson e = __copy_engines(i915_gem_context_engines(ctx)); 167710be98a7SChris Wilson mutex_unlock(&ctx->engines_mutex); 167810be98a7SChris Wilson if (IS_ERR_OR_NULL(e)) { 167910be98a7SChris Wilson args->size = 0; 168010be98a7SChris Wilson return PTR_ERR_OR_ZERO(e); 168110be98a7SChris Wilson } 168210be98a7SChris Wilson 168310be98a7SChris Wilson count = e->num_engines; 168410be98a7SChris Wilson 168510be98a7SChris Wilson /* Be paranoid in case we have an impedance mismatch */ 168610be98a7SChris Wilson if (!check_struct_size(user, engines, count, &size)) { 168710be98a7SChris Wilson err = -EINVAL; 168810be98a7SChris Wilson goto err_free; 168910be98a7SChris Wilson } 169010be98a7SChris Wilson if (overflows_type(size, args->size)) { 169110be98a7SChris Wilson err = -EINVAL; 169210be98a7SChris Wilson goto err_free; 169310be98a7SChris Wilson } 169410be98a7SChris Wilson 169510be98a7SChris Wilson if (!args->size) { 169610be98a7SChris Wilson args->size = size; 169710be98a7SChris Wilson goto err_free; 169810be98a7SChris Wilson } 169910be98a7SChris Wilson 170010be98a7SChris Wilson if (args->size < size) { 170110be98a7SChris Wilson err = -EINVAL; 170210be98a7SChris Wilson goto err_free; 170310be98a7SChris Wilson } 170410be98a7SChris Wilson 170510be98a7SChris Wilson user = u64_to_user_ptr(args->value); 170610be98a7SChris Wilson if (!access_ok(user, size)) { 170710be98a7SChris Wilson err = -EFAULT; 170810be98a7SChris Wilson goto err_free; 170910be98a7SChris Wilson } 171010be98a7SChris Wilson 171110be98a7SChris Wilson if (put_user(0, &user->extensions)) { 171210be98a7SChris Wilson err = -EFAULT; 171310be98a7SChris Wilson goto err_free; 171410be98a7SChris Wilson } 171510be98a7SChris Wilson 171610be98a7SChris Wilson for (n = 0; n < count; n++) { 171710be98a7SChris Wilson struct i915_engine_class_instance ci = { 171810be98a7SChris Wilson .engine_class = I915_ENGINE_CLASS_INVALID, 171910be98a7SChris Wilson .engine_instance = I915_ENGINE_CLASS_INVALID_NONE, 172010be98a7SChris Wilson }; 172110be98a7SChris Wilson 172210be98a7SChris Wilson if (e->engines[n]) { 172310be98a7SChris Wilson ci.engine_class = e->engines[n]->engine->uabi_class; 1724750e76b4SChris Wilson ci.engine_instance = e->engines[n]->engine->uabi_instance; 172510be98a7SChris Wilson } 172610be98a7SChris Wilson 172710be98a7SChris Wilson if (copy_to_user(&user->engines[n], &ci, sizeof(ci))) { 172810be98a7SChris Wilson err = -EFAULT; 172910be98a7SChris Wilson goto err_free; 173010be98a7SChris Wilson } 173110be98a7SChris Wilson } 173210be98a7SChris Wilson 173310be98a7SChris Wilson args->size = size; 173410be98a7SChris Wilson 173510be98a7SChris Wilson err_free: 1736155ab883SChris Wilson free_engines(e); 173710be98a7SChris Wilson return err; 173810be98a7SChris Wilson } 173910be98a7SChris Wilson 1740a0e04715SChris Wilson static int 1741a0e04715SChris Wilson set_persistence(struct i915_gem_context *ctx, 1742a0e04715SChris Wilson const struct drm_i915_gem_context_param *args) 1743a0e04715SChris Wilson { 1744a0e04715SChris Wilson if (args->size) 1745a0e04715SChris Wilson return -EINVAL; 1746a0e04715SChris Wilson 1747a0e04715SChris Wilson return __context_set_persistence(ctx, args->value); 1748a0e04715SChris Wilson } 1749a0e04715SChris Wilson 17500f100b70SChris Wilson static void __apply_priority(struct intel_context *ce, void *arg) 17510f100b70SChris Wilson { 17520f100b70SChris Wilson struct i915_gem_context *ctx = arg; 17530f100b70SChris Wilson 17540f100b70SChris Wilson if (!intel_engine_has_semaphores(ce->engine)) 17550f100b70SChris Wilson return; 17560f100b70SChris Wilson 17570f100b70SChris Wilson if (ctx->sched.priority >= I915_PRIORITY_NORMAL) 17580f100b70SChris Wilson intel_context_set_use_semaphores(ce); 17590f100b70SChris Wilson else 17600f100b70SChris Wilson intel_context_clear_use_semaphores(ce); 17610f100b70SChris Wilson } 17620f100b70SChris Wilson 17630f100b70SChris Wilson static int set_priority(struct i915_gem_context *ctx, 17640f100b70SChris Wilson const struct drm_i915_gem_context_param *args) 17650f100b70SChris Wilson { 17660f100b70SChris Wilson s64 priority = args->value; 17670f100b70SChris Wilson 17680f100b70SChris Wilson if (args->size) 17690f100b70SChris Wilson return -EINVAL; 17700f100b70SChris Wilson 17710f100b70SChris Wilson if (!(ctx->i915->caps.scheduler & I915_SCHEDULER_CAP_PRIORITY)) 17720f100b70SChris Wilson return -ENODEV; 17730f100b70SChris Wilson 17740f100b70SChris Wilson if (priority > I915_CONTEXT_MAX_USER_PRIORITY || 17750f100b70SChris Wilson priority < I915_CONTEXT_MIN_USER_PRIORITY) 17760f100b70SChris Wilson return -EINVAL; 17770f100b70SChris Wilson 17780f100b70SChris Wilson if (priority > I915_CONTEXT_DEFAULT_PRIORITY && 17790f100b70SChris Wilson !capable(CAP_SYS_NICE)) 17800f100b70SChris Wilson return -EPERM; 17810f100b70SChris Wilson 17820f100b70SChris Wilson ctx->sched.priority = I915_USER_PRIORITY(priority); 17830f100b70SChris Wilson context_apply_all(ctx, __apply_priority, ctx); 17840f100b70SChris Wilson 17850f100b70SChris Wilson return 0; 17860f100b70SChris Wilson } 17870f100b70SChris Wilson 178810be98a7SChris Wilson static int ctx_setparam(struct drm_i915_file_private *fpriv, 178910be98a7SChris Wilson struct i915_gem_context *ctx, 179010be98a7SChris Wilson struct drm_i915_gem_context_param *args) 179110be98a7SChris Wilson { 179210be98a7SChris Wilson int ret = 0; 179310be98a7SChris Wilson 179410be98a7SChris Wilson switch (args->param) { 179510be98a7SChris Wilson case I915_CONTEXT_PARAM_NO_ZEROMAP: 179610be98a7SChris Wilson if (args->size) 179710be98a7SChris Wilson ret = -EINVAL; 179810be98a7SChris Wilson else if (args->value) 179910be98a7SChris Wilson set_bit(UCONTEXT_NO_ZEROMAP, &ctx->user_flags); 180010be98a7SChris Wilson else 180110be98a7SChris Wilson clear_bit(UCONTEXT_NO_ZEROMAP, &ctx->user_flags); 180210be98a7SChris Wilson break; 180310be98a7SChris Wilson 180410be98a7SChris Wilson case I915_CONTEXT_PARAM_NO_ERROR_CAPTURE: 180510be98a7SChris Wilson if (args->size) 180610be98a7SChris Wilson ret = -EINVAL; 180710be98a7SChris Wilson else if (args->value) 180810be98a7SChris Wilson i915_gem_context_set_no_error_capture(ctx); 180910be98a7SChris Wilson else 181010be98a7SChris Wilson i915_gem_context_clear_no_error_capture(ctx); 181110be98a7SChris Wilson break; 181210be98a7SChris Wilson 181310be98a7SChris Wilson case I915_CONTEXT_PARAM_BANNABLE: 181410be98a7SChris Wilson if (args->size) 181510be98a7SChris Wilson ret = -EINVAL; 181610be98a7SChris Wilson else if (!capable(CAP_SYS_ADMIN) && !args->value) 181710be98a7SChris Wilson ret = -EPERM; 181810be98a7SChris Wilson else if (args->value) 181910be98a7SChris Wilson i915_gem_context_set_bannable(ctx); 182010be98a7SChris Wilson else 182110be98a7SChris Wilson i915_gem_context_clear_bannable(ctx); 182210be98a7SChris Wilson break; 182310be98a7SChris Wilson 182410be98a7SChris Wilson case I915_CONTEXT_PARAM_RECOVERABLE: 182510be98a7SChris Wilson if (args->size) 182610be98a7SChris Wilson ret = -EINVAL; 182710be98a7SChris Wilson else if (args->value) 182810be98a7SChris Wilson i915_gem_context_set_recoverable(ctx); 182910be98a7SChris Wilson else 183010be98a7SChris Wilson i915_gem_context_clear_recoverable(ctx); 183110be98a7SChris Wilson break; 183210be98a7SChris Wilson 183310be98a7SChris Wilson case I915_CONTEXT_PARAM_PRIORITY: 18340f100b70SChris Wilson ret = set_priority(ctx, args); 183510be98a7SChris Wilson break; 183610be98a7SChris Wilson 183710be98a7SChris Wilson case I915_CONTEXT_PARAM_SSEU: 183810be98a7SChris Wilson ret = set_sseu(ctx, args); 183910be98a7SChris Wilson break; 184010be98a7SChris Wilson 184110be98a7SChris Wilson case I915_CONTEXT_PARAM_VM: 184210be98a7SChris Wilson ret = set_ppgtt(fpriv, ctx, args); 184310be98a7SChris Wilson break; 184410be98a7SChris Wilson 184510be98a7SChris Wilson case I915_CONTEXT_PARAM_ENGINES: 184610be98a7SChris Wilson ret = set_engines(ctx, args); 184710be98a7SChris Wilson break; 184810be98a7SChris Wilson 1849a0e04715SChris Wilson case I915_CONTEXT_PARAM_PERSISTENCE: 1850a0e04715SChris Wilson ret = set_persistence(ctx, args); 1851a0e04715SChris Wilson break; 1852a0e04715SChris Wilson 185310be98a7SChris Wilson case I915_CONTEXT_PARAM_BAN_PERIOD: 185410be98a7SChris Wilson default: 185510be98a7SChris Wilson ret = -EINVAL; 185610be98a7SChris Wilson break; 185710be98a7SChris Wilson } 185810be98a7SChris Wilson 185910be98a7SChris Wilson return ret; 186010be98a7SChris Wilson } 186110be98a7SChris Wilson 186210be98a7SChris Wilson struct create_ext { 186310be98a7SChris Wilson struct i915_gem_context *ctx; 186410be98a7SChris Wilson struct drm_i915_file_private *fpriv; 186510be98a7SChris Wilson }; 186610be98a7SChris Wilson 186710be98a7SChris Wilson static int create_setparam(struct i915_user_extension __user *ext, void *data) 186810be98a7SChris Wilson { 186910be98a7SChris Wilson struct drm_i915_gem_context_create_ext_setparam local; 187010be98a7SChris Wilson const struct create_ext *arg = data; 187110be98a7SChris Wilson 187210be98a7SChris Wilson if (copy_from_user(&local, ext, sizeof(local))) 187310be98a7SChris Wilson return -EFAULT; 187410be98a7SChris Wilson 187510be98a7SChris Wilson if (local.param.ctx_id) 187610be98a7SChris Wilson return -EINVAL; 187710be98a7SChris Wilson 187810be98a7SChris Wilson return ctx_setparam(arg->fpriv, arg->ctx, &local.param); 187910be98a7SChris Wilson } 188010be98a7SChris Wilson 188110be98a7SChris Wilson static int clone_engines(struct i915_gem_context *dst, 188210be98a7SChris Wilson struct i915_gem_context *src) 188310be98a7SChris Wilson { 188410be98a7SChris Wilson struct i915_gem_engines *e = i915_gem_context_lock_engines(src); 188510be98a7SChris Wilson struct i915_gem_engines *clone; 188610be98a7SChris Wilson bool user_engines; 188710be98a7SChris Wilson unsigned long n; 188810be98a7SChris Wilson 188910be98a7SChris Wilson clone = kmalloc(struct_size(e, engines, e->num_engines), GFP_KERNEL); 189010be98a7SChris Wilson if (!clone) 189110be98a7SChris Wilson goto err_unlock; 189210be98a7SChris Wilson 1893155ab883SChris Wilson init_rcu_head(&clone->rcu); 189410be98a7SChris Wilson for (n = 0; n < e->num_engines; n++) { 189510be98a7SChris Wilson struct intel_engine_cs *engine; 189610be98a7SChris Wilson 189710be98a7SChris Wilson if (!e->engines[n]) { 189810be98a7SChris Wilson clone->engines[n] = NULL; 189910be98a7SChris Wilson continue; 190010be98a7SChris Wilson } 190110be98a7SChris Wilson engine = e->engines[n]->engine; 190210be98a7SChris Wilson 190310be98a7SChris Wilson /* 190410be98a7SChris Wilson * Virtual engines are singletons; they can only exist 190510be98a7SChris Wilson * inside a single context, because they embed their 190610be98a7SChris Wilson * HW context... As each virtual context implies a single 190710be98a7SChris Wilson * timeline (each engine can only dequeue a single request 190810be98a7SChris Wilson * at any time), it would be surprising for two contexts 190910be98a7SChris Wilson * to use the same engine. So let's create a copy of 191010be98a7SChris Wilson * the virtual engine instead. 191110be98a7SChris Wilson */ 191210be98a7SChris Wilson if (intel_engine_is_virtual(engine)) 191310be98a7SChris Wilson clone->engines[n] = 1914e6ba7648SChris Wilson intel_execlists_clone_virtual(engine); 191510be98a7SChris Wilson else 1916e6ba7648SChris Wilson clone->engines[n] = intel_context_create(engine); 191710be98a7SChris Wilson if (IS_ERR_OR_NULL(clone->engines[n])) { 191810be98a7SChris Wilson __free_engines(clone, n); 191910be98a7SChris Wilson goto err_unlock; 192010be98a7SChris Wilson } 1921e6ba7648SChris Wilson 1922e6ba7648SChris Wilson intel_context_set_gem(clone->engines[n], dst); 192310be98a7SChris Wilson } 192410be98a7SChris Wilson clone->num_engines = n; 192510be98a7SChris Wilson 192610be98a7SChris Wilson user_engines = i915_gem_context_user_engines(src); 192710be98a7SChris Wilson i915_gem_context_unlock_engines(src); 192810be98a7SChris Wilson 1929d96bb620SChris Wilson /* Serialised by constructor */ 1930d96bb620SChris Wilson free_engines(__context_engines_static(dst)); 193110be98a7SChris Wilson RCU_INIT_POINTER(dst->engines, clone); 193210be98a7SChris Wilson if (user_engines) 193310be98a7SChris Wilson i915_gem_context_set_user_engines(dst); 193410be98a7SChris Wilson else 193510be98a7SChris Wilson i915_gem_context_clear_user_engines(dst); 193610be98a7SChris Wilson return 0; 193710be98a7SChris Wilson 193810be98a7SChris Wilson err_unlock: 193910be98a7SChris Wilson i915_gem_context_unlock_engines(src); 194010be98a7SChris Wilson return -ENOMEM; 194110be98a7SChris Wilson } 194210be98a7SChris Wilson 194310be98a7SChris Wilson static int clone_flags(struct i915_gem_context *dst, 194410be98a7SChris Wilson struct i915_gem_context *src) 194510be98a7SChris Wilson { 194610be98a7SChris Wilson dst->user_flags = src->user_flags; 194710be98a7SChris Wilson return 0; 194810be98a7SChris Wilson } 194910be98a7SChris Wilson 195010be98a7SChris Wilson static int clone_schedattr(struct i915_gem_context *dst, 195110be98a7SChris Wilson struct i915_gem_context *src) 195210be98a7SChris Wilson { 195310be98a7SChris Wilson dst->sched = src->sched; 195410be98a7SChris Wilson return 0; 195510be98a7SChris Wilson } 195610be98a7SChris Wilson 195710be98a7SChris Wilson static int clone_sseu(struct i915_gem_context *dst, 195810be98a7SChris Wilson struct i915_gem_context *src) 195910be98a7SChris Wilson { 196010be98a7SChris Wilson struct i915_gem_engines *e = i915_gem_context_lock_engines(src); 196110be98a7SChris Wilson struct i915_gem_engines *clone; 196210be98a7SChris Wilson unsigned long n; 196310be98a7SChris Wilson int err; 196410be98a7SChris Wilson 1965d96bb620SChris Wilson /* no locking required; sole access under constructor*/ 1966d96bb620SChris Wilson clone = __context_engines_static(dst); 196710be98a7SChris Wilson if (e->num_engines != clone->num_engines) { 196810be98a7SChris Wilson err = -EINVAL; 196910be98a7SChris Wilson goto unlock; 197010be98a7SChris Wilson } 197110be98a7SChris Wilson 197210be98a7SChris Wilson for (n = 0; n < e->num_engines; n++) { 197310be98a7SChris Wilson struct intel_context *ce = e->engines[n]; 197410be98a7SChris Wilson 197510be98a7SChris Wilson if (clone->engines[n]->engine->class != ce->engine->class) { 197610be98a7SChris Wilson /* Must have compatible engine maps! */ 197710be98a7SChris Wilson err = -EINVAL; 197810be98a7SChris Wilson goto unlock; 197910be98a7SChris Wilson } 198010be98a7SChris Wilson 198110be98a7SChris Wilson /* serialises with set_sseu */ 198210be98a7SChris Wilson err = intel_context_lock_pinned(ce); 198310be98a7SChris Wilson if (err) 198410be98a7SChris Wilson goto unlock; 198510be98a7SChris Wilson 198610be98a7SChris Wilson clone->engines[n]->sseu = ce->sseu; 198710be98a7SChris Wilson intel_context_unlock_pinned(ce); 198810be98a7SChris Wilson } 198910be98a7SChris Wilson 199010be98a7SChris Wilson err = 0; 199110be98a7SChris Wilson unlock: 199210be98a7SChris Wilson i915_gem_context_unlock_engines(src); 199310be98a7SChris Wilson return err; 199410be98a7SChris Wilson } 199510be98a7SChris Wilson 199610be98a7SChris Wilson static int clone_timeline(struct i915_gem_context *dst, 199710be98a7SChris Wilson struct i915_gem_context *src) 199810be98a7SChris Wilson { 199975d0a7f3SChris Wilson if (src->timeline) 200075d0a7f3SChris Wilson __assign_timeline(dst, src->timeline); 200110be98a7SChris Wilson 200210be98a7SChris Wilson return 0; 200310be98a7SChris Wilson } 200410be98a7SChris Wilson 200510be98a7SChris Wilson static int clone_vm(struct i915_gem_context *dst, 200610be98a7SChris Wilson struct i915_gem_context *src) 200710be98a7SChris Wilson { 2008e568ac38SChris Wilson struct i915_address_space *vm; 2009a4e7ccdaSChris Wilson int err = 0; 201010be98a7SChris Wilson 201127dbae8fSChris Wilson if (!rcu_access_pointer(src->vm)) 201227dbae8fSChris Wilson return 0; 201327dbae8fSChris Wilson 201410be98a7SChris Wilson rcu_read_lock(); 201527dbae8fSChris Wilson vm = context_get_vm_rcu(src); 201610be98a7SChris Wilson rcu_read_unlock(); 201710be98a7SChris Wilson 2018a4e7ccdaSChris Wilson if (!mutex_lock_interruptible(&dst->mutex)) { 2019e568ac38SChris Wilson __assign_ppgtt(dst, vm); 2020a4e7ccdaSChris Wilson mutex_unlock(&dst->mutex); 2021a4e7ccdaSChris Wilson } else { 2022a4e7ccdaSChris Wilson err = -EINTR; 2023a4e7ccdaSChris Wilson } 202410be98a7SChris Wilson 202527dbae8fSChris Wilson i915_vm_put(vm); 2026a4e7ccdaSChris Wilson return err; 202710be98a7SChris Wilson } 202810be98a7SChris Wilson 202910be98a7SChris Wilson static int create_clone(struct i915_user_extension __user *ext, void *data) 203010be98a7SChris Wilson { 203110be98a7SChris Wilson static int (* const fn[])(struct i915_gem_context *dst, 203210be98a7SChris Wilson struct i915_gem_context *src) = { 203310be98a7SChris Wilson #define MAP(x, y) [ilog2(I915_CONTEXT_CLONE_##x)] = y 203410be98a7SChris Wilson MAP(ENGINES, clone_engines), 203510be98a7SChris Wilson MAP(FLAGS, clone_flags), 203610be98a7SChris Wilson MAP(SCHEDATTR, clone_schedattr), 203710be98a7SChris Wilson MAP(SSEU, clone_sseu), 203810be98a7SChris Wilson MAP(TIMELINE, clone_timeline), 203910be98a7SChris Wilson MAP(VM, clone_vm), 204010be98a7SChris Wilson #undef MAP 204110be98a7SChris Wilson }; 204210be98a7SChris Wilson struct drm_i915_gem_context_create_ext_clone local; 204310be98a7SChris Wilson const struct create_ext *arg = data; 204410be98a7SChris Wilson struct i915_gem_context *dst = arg->ctx; 204510be98a7SChris Wilson struct i915_gem_context *src; 204610be98a7SChris Wilson int err, bit; 204710be98a7SChris Wilson 204810be98a7SChris Wilson if (copy_from_user(&local, ext, sizeof(local))) 204910be98a7SChris Wilson return -EFAULT; 205010be98a7SChris Wilson 205110be98a7SChris Wilson BUILD_BUG_ON(GENMASK(BITS_PER_TYPE(local.flags) - 1, ARRAY_SIZE(fn)) != 205210be98a7SChris Wilson I915_CONTEXT_CLONE_UNKNOWN); 205310be98a7SChris Wilson 205410be98a7SChris Wilson if (local.flags & I915_CONTEXT_CLONE_UNKNOWN) 205510be98a7SChris Wilson return -EINVAL; 205610be98a7SChris Wilson 205710be98a7SChris Wilson if (local.rsvd) 205810be98a7SChris Wilson return -EINVAL; 205910be98a7SChris Wilson 206010be98a7SChris Wilson rcu_read_lock(); 206110be98a7SChris Wilson src = __i915_gem_context_lookup_rcu(arg->fpriv, local.clone_id); 206210be98a7SChris Wilson rcu_read_unlock(); 206310be98a7SChris Wilson if (!src) 206410be98a7SChris Wilson return -ENOENT; 206510be98a7SChris Wilson 206610be98a7SChris Wilson GEM_BUG_ON(src == dst); 206710be98a7SChris Wilson 206810be98a7SChris Wilson for (bit = 0; bit < ARRAY_SIZE(fn); bit++) { 206910be98a7SChris Wilson if (!(local.flags & BIT(bit))) 207010be98a7SChris Wilson continue; 207110be98a7SChris Wilson 207210be98a7SChris Wilson err = fn[bit](dst, src); 207310be98a7SChris Wilson if (err) 207410be98a7SChris Wilson return err; 207510be98a7SChris Wilson } 207610be98a7SChris Wilson 207710be98a7SChris Wilson return 0; 207810be98a7SChris Wilson } 207910be98a7SChris Wilson 208010be98a7SChris Wilson static const i915_user_extension_fn create_extensions[] = { 208110be98a7SChris Wilson [I915_CONTEXT_CREATE_EXT_SETPARAM] = create_setparam, 208210be98a7SChris Wilson [I915_CONTEXT_CREATE_EXT_CLONE] = create_clone, 208310be98a7SChris Wilson }; 208410be98a7SChris Wilson 208510be98a7SChris Wilson static bool client_is_banned(struct drm_i915_file_private *file_priv) 208610be98a7SChris Wilson { 208710be98a7SChris Wilson return atomic_read(&file_priv->ban_score) >= I915_CLIENT_SCORE_BANNED; 208810be98a7SChris Wilson } 208910be98a7SChris Wilson 209010be98a7SChris Wilson int i915_gem_context_create_ioctl(struct drm_device *dev, void *data, 209110be98a7SChris Wilson struct drm_file *file) 209210be98a7SChris Wilson { 209310be98a7SChris Wilson struct drm_i915_private *i915 = to_i915(dev); 209410be98a7SChris Wilson struct drm_i915_gem_context_create_ext *args = data; 209510be98a7SChris Wilson struct create_ext ext_data; 209610be98a7SChris Wilson int ret; 2097c100777cSTvrtko Ursulin u32 id; 209810be98a7SChris Wilson 209910be98a7SChris Wilson if (!DRIVER_CAPS(i915)->has_logical_contexts) 210010be98a7SChris Wilson return -ENODEV; 210110be98a7SChris Wilson 210210be98a7SChris Wilson if (args->flags & I915_CONTEXT_CREATE_FLAGS_UNKNOWN) 210310be98a7SChris Wilson return -EINVAL; 210410be98a7SChris Wilson 2105cb823ed9SChris Wilson ret = intel_gt_terminally_wedged(&i915->gt); 210610be98a7SChris Wilson if (ret) 210710be98a7SChris Wilson return ret; 210810be98a7SChris Wilson 210910be98a7SChris Wilson ext_data.fpriv = file->driver_priv; 211010be98a7SChris Wilson if (client_is_banned(ext_data.fpriv)) { 2111baa89ba3SWambui Karuga drm_dbg(&i915->drm, 2112baa89ba3SWambui Karuga "client %s[%d] banned from creating ctx\n", 2113ba16a48aSTvrtko Ursulin current->comm, task_pid_nr(current)); 211410be98a7SChris Wilson return -EIO; 211510be98a7SChris Wilson } 211610be98a7SChris Wilson 211710be98a7SChris Wilson ext_data.ctx = i915_gem_create_context(i915, args->flags); 211810be98a7SChris Wilson if (IS_ERR(ext_data.ctx)) 211910be98a7SChris Wilson return PTR_ERR(ext_data.ctx); 212010be98a7SChris Wilson 212110be98a7SChris Wilson if (args->flags & I915_CONTEXT_CREATE_FLAGS_USE_EXTENSIONS) { 212210be98a7SChris Wilson ret = i915_user_extensions(u64_to_user_ptr(args->extensions), 212310be98a7SChris Wilson create_extensions, 212410be98a7SChris Wilson ARRAY_SIZE(create_extensions), 212510be98a7SChris Wilson &ext_data); 212610be98a7SChris Wilson if (ret) 212710be98a7SChris Wilson goto err_ctx; 212810be98a7SChris Wilson } 212910be98a7SChris Wilson 2130c100777cSTvrtko Ursulin ret = gem_context_register(ext_data.ctx, ext_data.fpriv, &id); 213110be98a7SChris Wilson if (ret < 0) 213210be98a7SChris Wilson goto err_ctx; 213310be98a7SChris Wilson 2134c100777cSTvrtko Ursulin args->ctx_id = id; 2135baa89ba3SWambui Karuga drm_dbg(&i915->drm, "HW context %d created\n", args->ctx_id); 213610be98a7SChris Wilson 213710be98a7SChris Wilson return 0; 213810be98a7SChris Wilson 213910be98a7SChris Wilson err_ctx: 214010be98a7SChris Wilson context_close(ext_data.ctx); 214110be98a7SChris Wilson return ret; 214210be98a7SChris Wilson } 214310be98a7SChris Wilson 214410be98a7SChris Wilson int i915_gem_context_destroy_ioctl(struct drm_device *dev, void *data, 214510be98a7SChris Wilson struct drm_file *file) 214610be98a7SChris Wilson { 214710be98a7SChris Wilson struct drm_i915_gem_context_destroy *args = data; 214810be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 214910be98a7SChris Wilson struct i915_gem_context *ctx; 215010be98a7SChris Wilson 215110be98a7SChris Wilson if (args->pad != 0) 215210be98a7SChris Wilson return -EINVAL; 215310be98a7SChris Wilson 215410be98a7SChris Wilson if (!args->ctx_id) 215510be98a7SChris Wilson return -ENOENT; 215610be98a7SChris Wilson 2157c100777cSTvrtko Ursulin ctx = xa_erase(&file_priv->context_xa, args->ctx_id); 215810be98a7SChris Wilson if (!ctx) 215910be98a7SChris Wilson return -ENOENT; 216010be98a7SChris Wilson 216110be98a7SChris Wilson context_close(ctx); 216210be98a7SChris Wilson return 0; 216310be98a7SChris Wilson } 216410be98a7SChris Wilson 216510be98a7SChris Wilson static int get_sseu(struct i915_gem_context *ctx, 216610be98a7SChris Wilson struct drm_i915_gem_context_param *args) 216710be98a7SChris Wilson { 216810be98a7SChris Wilson struct drm_i915_gem_context_param_sseu user_sseu; 216910be98a7SChris Wilson struct intel_context *ce; 217010be98a7SChris Wilson unsigned long lookup; 217110be98a7SChris Wilson int err; 217210be98a7SChris Wilson 217310be98a7SChris Wilson if (args->size == 0) 217410be98a7SChris Wilson goto out; 217510be98a7SChris Wilson else if (args->size < sizeof(user_sseu)) 217610be98a7SChris Wilson return -EINVAL; 217710be98a7SChris Wilson 217810be98a7SChris Wilson if (copy_from_user(&user_sseu, u64_to_user_ptr(args->value), 217910be98a7SChris Wilson sizeof(user_sseu))) 218010be98a7SChris Wilson return -EFAULT; 218110be98a7SChris Wilson 218210be98a7SChris Wilson if (user_sseu.rsvd) 218310be98a7SChris Wilson return -EINVAL; 218410be98a7SChris Wilson 218510be98a7SChris Wilson if (user_sseu.flags & ~(I915_CONTEXT_SSEU_FLAG_ENGINE_INDEX)) 218610be98a7SChris Wilson return -EINVAL; 218710be98a7SChris Wilson 218810be98a7SChris Wilson lookup = 0; 218910be98a7SChris Wilson if (user_sseu.flags & I915_CONTEXT_SSEU_FLAG_ENGINE_INDEX) 219010be98a7SChris Wilson lookup |= LOOKUP_USER_INDEX; 219110be98a7SChris Wilson 219210be98a7SChris Wilson ce = lookup_user_engine(ctx, lookup, &user_sseu.engine); 219310be98a7SChris Wilson if (IS_ERR(ce)) 219410be98a7SChris Wilson return PTR_ERR(ce); 219510be98a7SChris Wilson 219610be98a7SChris Wilson err = intel_context_lock_pinned(ce); /* serialises with set_sseu */ 219710be98a7SChris Wilson if (err) { 219810be98a7SChris Wilson intel_context_put(ce); 219910be98a7SChris Wilson return err; 220010be98a7SChris Wilson } 220110be98a7SChris Wilson 220210be98a7SChris Wilson user_sseu.slice_mask = ce->sseu.slice_mask; 220310be98a7SChris Wilson user_sseu.subslice_mask = ce->sseu.subslice_mask; 220410be98a7SChris Wilson user_sseu.min_eus_per_subslice = ce->sseu.min_eus_per_subslice; 220510be98a7SChris Wilson user_sseu.max_eus_per_subslice = ce->sseu.max_eus_per_subslice; 220610be98a7SChris Wilson 220710be98a7SChris Wilson intel_context_unlock_pinned(ce); 220810be98a7SChris Wilson intel_context_put(ce); 220910be98a7SChris Wilson 221010be98a7SChris Wilson if (copy_to_user(u64_to_user_ptr(args->value), &user_sseu, 221110be98a7SChris Wilson sizeof(user_sseu))) 221210be98a7SChris Wilson return -EFAULT; 221310be98a7SChris Wilson 221410be98a7SChris Wilson out: 221510be98a7SChris Wilson args->size = sizeof(user_sseu); 221610be98a7SChris Wilson 221710be98a7SChris Wilson return 0; 221810be98a7SChris Wilson } 221910be98a7SChris Wilson 222010be98a7SChris Wilson int i915_gem_context_getparam_ioctl(struct drm_device *dev, void *data, 222110be98a7SChris Wilson struct drm_file *file) 222210be98a7SChris Wilson { 222310be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 222410be98a7SChris Wilson struct drm_i915_gem_context_param *args = data; 222510be98a7SChris Wilson struct i915_gem_context *ctx; 222610be98a7SChris Wilson int ret = 0; 222710be98a7SChris Wilson 222810be98a7SChris Wilson ctx = i915_gem_context_lookup(file_priv, args->ctx_id); 222910be98a7SChris Wilson if (!ctx) 223010be98a7SChris Wilson return -ENOENT; 223110be98a7SChris Wilson 223210be98a7SChris Wilson switch (args->param) { 223310be98a7SChris Wilson case I915_CONTEXT_PARAM_NO_ZEROMAP: 223410be98a7SChris Wilson args->size = 0; 223510be98a7SChris Wilson args->value = test_bit(UCONTEXT_NO_ZEROMAP, &ctx->user_flags); 223610be98a7SChris Wilson break; 223710be98a7SChris Wilson 223810be98a7SChris Wilson case I915_CONTEXT_PARAM_GTT_SIZE: 223910be98a7SChris Wilson args->size = 0; 2240a4e7ccdaSChris Wilson rcu_read_lock(); 2241a4e7ccdaSChris Wilson if (rcu_access_pointer(ctx->vm)) 2242a4e7ccdaSChris Wilson args->value = rcu_dereference(ctx->vm)->total; 224310be98a7SChris Wilson else 224410be98a7SChris Wilson args->value = to_i915(dev)->ggtt.vm.total; 2245a4e7ccdaSChris Wilson rcu_read_unlock(); 224610be98a7SChris Wilson break; 224710be98a7SChris Wilson 224810be98a7SChris Wilson case I915_CONTEXT_PARAM_NO_ERROR_CAPTURE: 224910be98a7SChris Wilson args->size = 0; 225010be98a7SChris Wilson args->value = i915_gem_context_no_error_capture(ctx); 225110be98a7SChris Wilson break; 225210be98a7SChris Wilson 225310be98a7SChris Wilson case I915_CONTEXT_PARAM_BANNABLE: 225410be98a7SChris Wilson args->size = 0; 225510be98a7SChris Wilson args->value = i915_gem_context_is_bannable(ctx); 225610be98a7SChris Wilson break; 225710be98a7SChris Wilson 225810be98a7SChris Wilson case I915_CONTEXT_PARAM_RECOVERABLE: 225910be98a7SChris Wilson args->size = 0; 226010be98a7SChris Wilson args->value = i915_gem_context_is_recoverable(ctx); 226110be98a7SChris Wilson break; 226210be98a7SChris Wilson 226310be98a7SChris Wilson case I915_CONTEXT_PARAM_PRIORITY: 226410be98a7SChris Wilson args->size = 0; 226510be98a7SChris Wilson args->value = ctx->sched.priority >> I915_USER_PRIORITY_SHIFT; 226610be98a7SChris Wilson break; 226710be98a7SChris Wilson 226810be98a7SChris Wilson case I915_CONTEXT_PARAM_SSEU: 226910be98a7SChris Wilson ret = get_sseu(ctx, args); 227010be98a7SChris Wilson break; 227110be98a7SChris Wilson 227210be98a7SChris Wilson case I915_CONTEXT_PARAM_VM: 227310be98a7SChris Wilson ret = get_ppgtt(file_priv, ctx, args); 227410be98a7SChris Wilson break; 227510be98a7SChris Wilson 227610be98a7SChris Wilson case I915_CONTEXT_PARAM_ENGINES: 227710be98a7SChris Wilson ret = get_engines(ctx, args); 227810be98a7SChris Wilson break; 227910be98a7SChris Wilson 2280a0e04715SChris Wilson case I915_CONTEXT_PARAM_PERSISTENCE: 2281a0e04715SChris Wilson args->size = 0; 2282a0e04715SChris Wilson args->value = i915_gem_context_is_persistent(ctx); 2283a0e04715SChris Wilson break; 2284a0e04715SChris Wilson 228510be98a7SChris Wilson case I915_CONTEXT_PARAM_BAN_PERIOD: 228610be98a7SChris Wilson default: 228710be98a7SChris Wilson ret = -EINVAL; 228810be98a7SChris Wilson break; 228910be98a7SChris Wilson } 229010be98a7SChris Wilson 229110be98a7SChris Wilson i915_gem_context_put(ctx); 229210be98a7SChris Wilson return ret; 229310be98a7SChris Wilson } 229410be98a7SChris Wilson 229510be98a7SChris Wilson int i915_gem_context_setparam_ioctl(struct drm_device *dev, void *data, 229610be98a7SChris Wilson struct drm_file *file) 229710be98a7SChris Wilson { 229810be98a7SChris Wilson struct drm_i915_file_private *file_priv = file->driver_priv; 229910be98a7SChris Wilson struct drm_i915_gem_context_param *args = data; 230010be98a7SChris Wilson struct i915_gem_context *ctx; 230110be98a7SChris Wilson int ret; 230210be98a7SChris Wilson 230310be98a7SChris Wilson ctx = i915_gem_context_lookup(file_priv, args->ctx_id); 230410be98a7SChris Wilson if (!ctx) 230510be98a7SChris Wilson return -ENOENT; 230610be98a7SChris Wilson 230710be98a7SChris Wilson ret = ctx_setparam(file_priv, ctx, args); 230810be98a7SChris Wilson 230910be98a7SChris Wilson i915_gem_context_put(ctx); 231010be98a7SChris Wilson return ret; 231110be98a7SChris Wilson } 231210be98a7SChris Wilson 231310be98a7SChris Wilson int i915_gem_context_reset_stats_ioctl(struct drm_device *dev, 231410be98a7SChris Wilson void *data, struct drm_file *file) 231510be98a7SChris Wilson { 2316a4e7ccdaSChris Wilson struct drm_i915_private *i915 = to_i915(dev); 231710be98a7SChris Wilson struct drm_i915_reset_stats *args = data; 231810be98a7SChris Wilson struct i915_gem_context *ctx; 231910be98a7SChris Wilson int ret; 232010be98a7SChris Wilson 232110be98a7SChris Wilson if (args->flags || args->pad) 232210be98a7SChris Wilson return -EINVAL; 232310be98a7SChris Wilson 232410be98a7SChris Wilson ret = -ENOENT; 232510be98a7SChris Wilson rcu_read_lock(); 232610be98a7SChris Wilson ctx = __i915_gem_context_lookup_rcu(file->driver_priv, args->ctx_id); 232710be98a7SChris Wilson if (!ctx) 232810be98a7SChris Wilson goto out; 232910be98a7SChris Wilson 233010be98a7SChris Wilson /* 233110be98a7SChris Wilson * We opt for unserialised reads here. This may result in tearing 233210be98a7SChris Wilson * in the extremely unlikely event of a GPU hang on this context 233310be98a7SChris Wilson * as we are querying them. If we need that extra layer of protection, 233410be98a7SChris Wilson * we should wrap the hangstats with a seqlock. 233510be98a7SChris Wilson */ 233610be98a7SChris Wilson 233710be98a7SChris Wilson if (capable(CAP_SYS_ADMIN)) 2338a4e7ccdaSChris Wilson args->reset_count = i915_reset_count(&i915->gpu_error); 233910be98a7SChris Wilson else 234010be98a7SChris Wilson args->reset_count = 0; 234110be98a7SChris Wilson 234210be98a7SChris Wilson args->batch_active = atomic_read(&ctx->guilty_count); 234310be98a7SChris Wilson args->batch_pending = atomic_read(&ctx->active_count); 234410be98a7SChris Wilson 234510be98a7SChris Wilson ret = 0; 234610be98a7SChris Wilson out: 234710be98a7SChris Wilson rcu_read_unlock(); 234810be98a7SChris Wilson return ret; 234910be98a7SChris Wilson } 235010be98a7SChris Wilson 235110be98a7SChris Wilson /* GEM context-engines iterator: for_each_gem_engine() */ 235210be98a7SChris Wilson struct intel_context * 235310be98a7SChris Wilson i915_gem_engines_iter_next(struct i915_gem_engines_iter *it) 235410be98a7SChris Wilson { 235510be98a7SChris Wilson const struct i915_gem_engines *e = it->engines; 235610be98a7SChris Wilson struct intel_context *ctx; 235710be98a7SChris Wilson 235810be98a7SChris Wilson do { 235910be98a7SChris Wilson if (it->idx >= e->num_engines) 236010be98a7SChris Wilson return NULL; 236110be98a7SChris Wilson 236210be98a7SChris Wilson ctx = e->engines[it->idx++]; 236310be98a7SChris Wilson } while (!ctx); 236410be98a7SChris Wilson 236510be98a7SChris Wilson return ctx; 236610be98a7SChris Wilson } 236710be98a7SChris Wilson 236810be98a7SChris Wilson #if IS_ENABLED(CONFIG_DRM_I915_SELFTEST) 236910be98a7SChris Wilson #include "selftests/mock_context.c" 237010be98a7SChris Wilson #include "selftests/i915_gem_context.c" 237110be98a7SChris Wilson #endif 237210be98a7SChris Wilson 237310be98a7SChris Wilson static void i915_global_gem_context_shrink(void) 237410be98a7SChris Wilson { 237510be98a7SChris Wilson kmem_cache_shrink(global.slab_luts); 237610be98a7SChris Wilson } 237710be98a7SChris Wilson 237810be98a7SChris Wilson static void i915_global_gem_context_exit(void) 237910be98a7SChris Wilson { 238010be98a7SChris Wilson kmem_cache_destroy(global.slab_luts); 238110be98a7SChris Wilson } 238210be98a7SChris Wilson 238310be98a7SChris Wilson static struct i915_global_gem_context global = { { 238410be98a7SChris Wilson .shrink = i915_global_gem_context_shrink, 238510be98a7SChris Wilson .exit = i915_global_gem_context_exit, 238610be98a7SChris Wilson } }; 238710be98a7SChris Wilson 238810be98a7SChris Wilson int __init i915_global_gem_context_init(void) 238910be98a7SChris Wilson { 239010be98a7SChris Wilson global.slab_luts = KMEM_CACHE(i915_lut_handle, 0); 239110be98a7SChris Wilson if (!global.slab_luts) 239210be98a7SChris Wilson return -ENOMEM; 239310be98a7SChris Wilson 239410be98a7SChris Wilson i915_global_register(&global.base); 239510be98a7SChris Wilson return 0; 239610be98a7SChris Wilson } 2397