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(filp: f.file, fpriv: &fpriv); |
50 | if (r) { |
51 | fdput(fd: 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, ctx_prio: priority); |
59 | mutex_unlock(lock: &mgr->lock); |
60 | |
61 | fdput(fd: 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(filp: f.file, fpriv: &fpriv); |
79 | if (r) { |
80 | fdput(fd: f); |
81 | return r; |
82 | } |
83 | |
84 | ctx = amdgpu_ctx_get(fpriv, id: ctx_id); |
85 | |
86 | if (!ctx) { |
87 | fdput(fd: f); |
88 | return -EINVAL; |
89 | } |
90 | |
91 | amdgpu_ctx_priority_override(ctx, ctx_prio: priority); |
92 | amdgpu_ctx_put(ctx); |
93 | fdput(fd: 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(ddev: 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(ctx_prio: 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 | fd: args->in.fd, |
125 | priority: args->in.priority); |
126 | break; |
127 | case AMDGPU_SCHED_OP_CONTEXT_PRIORITY_OVERRIDE: |
128 | r = amdgpu_sched_context_priority_override(adev, |
129 | fd: args->in.fd, |
130 | ctx_id: args->in.ctx_id, |
131 | priority: args->in.priority); |
132 | break; |
133 | default: |
134 | /* Impossible. |
135 | */ |
136 | r = -EINVAL; |
137 | break; |
138 | } |
139 | |
140 | return r; |
141 | } |
142 | |