1 /* 2 * Copyright 2017 Valve Corporation 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a 5 * copy of this software and associated documentation files (the "Software"), 6 * to deal in the Software without restriction, including without limitation 7 * the rights to use, copy, modify, merge, publish, distribute, sublicense, 8 * and/or sell copies of the Software, and to permit persons to whom the 9 * Software is furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice shall be included in 12 * all copies or substantial portions of the Software. 13 * 14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 17 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR 18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 20 * OTHER DEALINGS IN THE SOFTWARE. 21 * 22 * Authors: Andres Rodriguez <andresx7@gmail.com> 23 */ 24 25 #include <linux/fdtable.h> 26 #include <linux/file.h> 27 #include <linux/pid.h> 28 29 #include <drm/amdgpu_drm.h> 30 31 #include "amdgpu.h" 32 #include "amdgpu_sched.h" 33 #include "amdgpu_vm.h" 34 35 static int amdgpu_sched_process_priority_override(struct amdgpu_device *adev, 36 int fd, 37 int32_t priority) 38 { 39 struct fd f = fdget(fd); 40 struct amdgpu_fpriv *fpriv; 41 struct amdgpu_ctx_mgr *mgr; 42 struct amdgpu_ctx *ctx; 43 uint32_t id; 44 int r; 45 46 if (!f.file) 47 return -EINVAL; 48 49 r = amdgpu_file_to_fpriv(f.file, &fpriv); 50 if (r) { 51 fdput(f); 52 return r; 53 } 54 55 mgr = &fpriv->ctx_mgr; 56 mutex_lock(&mgr->lock); 57 idr_for_each_entry(&mgr->ctx_handles, ctx, id) 58 amdgpu_ctx_priority_override(ctx, priority); 59 mutex_unlock(&mgr->lock); 60 61 fdput(f); 62 return 0; 63 } 64 65 static int amdgpu_sched_context_priority_override(struct amdgpu_device *adev, 66 int fd, 67 unsigned ctx_id, 68 int32_t priority) 69 { 70 struct fd f = fdget(fd); 71 struct amdgpu_fpriv *fpriv; 72 struct amdgpu_ctx *ctx; 73 int r; 74 75 if (!f.file) 76 return -EINVAL; 77 78 r = amdgpu_file_to_fpriv(f.file, &fpriv); 79 if (r) { 80 fdput(f); 81 return r; 82 } 83 84 ctx = amdgpu_ctx_get(fpriv, ctx_id); 85 86 if (!ctx) { 87 fdput(f); 88 return -EINVAL; 89 } 90 91 amdgpu_ctx_priority_override(ctx, priority); 92 amdgpu_ctx_put(ctx); 93 fdput(f); 94 95 return 0; 96 } 97 98 int amdgpu_sched_ioctl(struct drm_device *dev, void *data, 99 struct drm_file *filp) 100 { 101 union drm_amdgpu_sched *args = data; 102 struct amdgpu_device *adev = drm_to_adev(dev); 103 int r; 104 105 /* First check the op, then the op's argument. 106 */ 107 switch (args->in.op) { 108 case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE: 109 case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE: 110 break; 111 default: 112 DRM_ERROR("Invalid sched op specified: %d\n", args->in.op); 113 return -EINVAL; 114 } 115 116 if (!amdgpu_ctx_priority_is_valid(args->in.priority)) { 117 WARN(1, "Invalid context priority %d\n", args->in.priority); 118 return -EINVAL; 119 } 120 121 switch (args->in.op) { 122 case AMDGPU_SCHED_OP_PROCESS_PRIORITY_OVERRIDE: 123 r = amdgpu_sched_process_priority_override(adev, 124 args->in.fd, 125 args->in.priority); 126 break; 127 case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE: 128 r = amdgpu_sched_context_priority_override(adev, 129 args->in.fd, 130 args->in.ctx_id, 131 args->in.priority); 132 break; 133 default: 134 /* Impossible. 135 */ 136 r = -EINVAL; 137 break; 138 } 139 140 return r; 141 } 142