Loading...
1/*
2 * Copyright 2008 Jerome Glisse.
3 * All Rights Reserved.
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
20 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
21 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *
24 * Authors:
25 * Jerome Glisse <glisse@freedesktop.org>
26 */
27
28#include <linux/file.h>
29#include <linux/pagemap.h>
30#include <linux/sync_file.h>
31#include <linux/dma-buf.h>
32
33#include <drm/amdgpu_drm.h>
34#include <drm/drm_syncobj.h>
35#include <drm/ttm/ttm_tt.h>
36
37#include "amdgpu_cs.h"
38#include "amdgpu.h"
39#include "amdgpu_trace.h"
40#include "amdgpu_gmc.h"
41#include "amdgpu_gem.h"
42#include "amdgpu_ras.h"
43
44static int amdgpu_cs_parser_init(struct amdgpu_cs_parser *p,
45 struct amdgpu_device *adev,
46 struct drm_file *filp,
47 union drm_amdgpu_cs *cs)
48{
49 struct amdgpu_fpriv *fpriv = filp->driver_priv;
50
51 if (cs->in.num_chunks == 0)
52 return -EINVAL;
53
54 memset(p, 0, sizeof(*p));
55 p->adev = adev;
56 p->filp = filp;
57
58 p->ctx = amdgpu_ctx_get(fpriv, cs->in.ctx_id);
59 if (!p->ctx)
60 return -EINVAL;
61
62 if (atomic_read(&p->ctx->guilty)) {
63 amdgpu_ctx_put(p->ctx);
64 return -ECANCELED;
65 }
66
67 amdgpu_sync_create(&p->sync);
68 drm_exec_init(&p->exec, DRM_EXEC_INTERRUPTIBLE_WAIT |
69 DRM_EXEC_IGNORE_DUPLICATES, 0);
70 return 0;
71}
72
73static int amdgpu_cs_job_idx(struct amdgpu_cs_parser *p,
74 struct drm_amdgpu_cs_chunk_ib *chunk_ib)
75{
76 struct drm_sched_entity *entity;
77 unsigned int i;
78 int r;
79
80 r = amdgpu_ctx_get_entity(p->ctx, chunk_ib->ip_type,
81 chunk_ib->ip_instance,
82 chunk_ib->ring, &entity);
83 if (r)
84 return r;
85
86 /*
87 * Abort if there is no run queue associated with this entity.
88 * Possibly because of disabled HW IP.
89 */
90 if (entity->rq == NULL)
91 return -EINVAL;
92
93 /* Check if we can add this IB to some existing job */
94 for (i = 0; i < p->gang_size; ++i)
95 if (p->entities[i] == entity)
96 return i;
97
98 /* If not increase the gang size if possible */
99 if (i == AMDGPU_CS_GANG_SIZE)
100 return -EINVAL;
101
102 p->entities[i] = entity;
103 p->gang_size = i + 1;
104 return i;
105}
106
107static int amdgpu_cs_p1_ib(struct amdgpu_cs_parser *p,
108 struct drm_amdgpu_cs_chunk_ib *chunk_ib,
109 unsigned int *num_ibs)
110{
111 int r;
112
113 r = amdgpu_cs_job_idx(p, chunk_ib);
114 if (r < 0)
115 return r;
116
117 if (num_ibs[r] >= amdgpu_ring_max_ibs(chunk_ib->ip_type))
118 return -EINVAL;
119
120 ++(num_ibs[r]);
121 p->gang_leader_idx = r;
122 return 0;
123}
124
125static int amdgpu_cs_p1_user_fence(struct amdgpu_cs_parser *p,
126 struct drm_amdgpu_cs_chunk_fence *data,
127 uint32_t *offset)
128{
129 struct drm_gem_object *gobj;
130 unsigned long size;
131
132 gobj = drm_gem_object_lookup(p->filp, data->handle);
133 if (gobj == NULL)
134 return -EINVAL;
135
136 p->uf_bo = amdgpu_bo_ref(gem_to_amdgpu_bo(gobj));
137 drm_gem_object_put(gobj);
138
139 size = amdgpu_bo_size(p->uf_bo);
140 if (size != PAGE_SIZE || data->offset > (size - 8))
141 return -EINVAL;
142
143 if (amdgpu_ttm_tt_get_usermm(p->uf_bo->tbo.ttm))
144 return -EINVAL;
145
146 *offset = data->offset;
147 return 0;
148}
149
150static int amdgpu_cs_p1_bo_handles(struct amdgpu_cs_parser *p,
151 struct drm_amdgpu_bo_list_in *data)
152{
153 struct drm_amdgpu_bo_list_entry *info;
154 int r;
155
156 r = amdgpu_bo_create_list_entry_array(data, &info);
157 if (r)
158 return r;
159
160 r = amdgpu_bo_list_create(p->adev, p->filp, info, data->bo_number,
161 &p->bo_list);
162 if (r)
163 goto error_free;
164
165 kvfree(info);
166 return 0;
167
168error_free:
169 kvfree(info);
170
171 return r;
172}
173
174/* Copy the data from userspace and go over it the first time */
175static int amdgpu_cs_pass1(struct amdgpu_cs_parser *p,
176 union drm_amdgpu_cs *cs)
177{
178 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
179 unsigned int num_ibs[AMDGPU_CS_GANG_SIZE] = { };
180 struct amdgpu_vm *vm = &fpriv->vm;
181 uint64_t *chunk_array_user;
182 uint64_t *chunk_array;
183 uint32_t uf_offset = 0;
184 size_t size;
185 int ret;
186 int i;
187
188 chunk_array = kvmalloc_array(cs->in.num_chunks, sizeof(uint64_t),
189 GFP_KERNEL);
190 if (!chunk_array)
191 return -ENOMEM;
192
193 /* get chunks */
194 chunk_array_user = u64_to_user_ptr(cs->in.chunks);
195 if (copy_from_user(chunk_array, chunk_array_user,
196 sizeof(uint64_t)*cs->in.num_chunks)) {
197 ret = -EFAULT;
198 goto free_chunk;
199 }
200
201 p->nchunks = cs->in.num_chunks;
202 p->chunks = kvmalloc_array(p->nchunks, sizeof(struct amdgpu_cs_chunk),
203 GFP_KERNEL);
204 if (!p->chunks) {
205 ret = -ENOMEM;
206 goto free_chunk;
207 }
208
209 for (i = 0; i < p->nchunks; i++) {
210 struct drm_amdgpu_cs_chunk __user *chunk_ptr = NULL;
211 struct drm_amdgpu_cs_chunk user_chunk;
212 uint32_t __user *cdata;
213
214 chunk_ptr = u64_to_user_ptr(chunk_array[i]);
215 if (copy_from_user(&user_chunk, chunk_ptr,
216 sizeof(struct drm_amdgpu_cs_chunk))) {
217 ret = -EFAULT;
218 i--;
219 goto free_partial_kdata;
220 }
221 p->chunks[i].chunk_id = user_chunk.chunk_id;
222 p->chunks[i].length_dw = user_chunk.length_dw;
223
224 size = p->chunks[i].length_dw;
225 cdata = u64_to_user_ptr(user_chunk.chunk_data);
226
227 p->chunks[i].kdata = kvmalloc_array(size, sizeof(uint32_t),
228 GFP_KERNEL);
229 if (p->chunks[i].kdata == NULL) {
230 ret = -ENOMEM;
231 i--;
232 goto free_partial_kdata;
233 }
234 size *= sizeof(uint32_t);
235 if (copy_from_user(p->chunks[i].kdata, cdata, size)) {
236 ret = -EFAULT;
237 goto free_partial_kdata;
238 }
239
240 /* Assume the worst on the following checks */
241 ret = -EINVAL;
242 switch (p->chunks[i].chunk_id) {
243 case AMDGPU_CHUNK_ID_IB:
244 if (size < sizeof(struct drm_amdgpu_cs_chunk_ib))
245 goto free_partial_kdata;
246
247 ret = amdgpu_cs_p1_ib(p, p->chunks[i].kdata, num_ibs);
248 if (ret)
249 goto free_partial_kdata;
250 break;
251
252 case AMDGPU_CHUNK_ID_FENCE:
253 if (size < sizeof(struct drm_amdgpu_cs_chunk_fence))
254 goto free_partial_kdata;
255
256 ret = amdgpu_cs_p1_user_fence(p, p->chunks[i].kdata,
257 &uf_offset);
258 if (ret)
259 goto free_partial_kdata;
260 break;
261
262 case AMDGPU_CHUNK_ID_BO_HANDLES:
263 if (size < sizeof(struct drm_amdgpu_bo_list_in))
264 goto free_partial_kdata;
265
266 ret = amdgpu_cs_p1_bo_handles(p, p->chunks[i].kdata);
267 if (ret)
268 goto free_partial_kdata;
269 break;
270
271 case AMDGPU_CHUNK_ID_DEPENDENCIES:
272 case AMDGPU_CHUNK_ID_SYNCOBJ_IN:
273 case AMDGPU_CHUNK_ID_SYNCOBJ_OUT:
274 case AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES:
275 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_WAIT:
276 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_SIGNAL:
277 case AMDGPU_CHUNK_ID_CP_GFX_SHADOW:
278 break;
279
280 default:
281 goto free_partial_kdata;
282 }
283 }
284
285 if (!p->gang_size) {
286 ret = -EINVAL;
287 goto free_all_kdata;
288 }
289
290 for (i = 0; i < p->gang_size; ++i) {
291 ret = amdgpu_job_alloc(p->adev, vm, p->entities[i], vm,
292 num_ibs[i], &p->jobs[i]);
293 if (ret)
294 goto free_all_kdata;
295 }
296 p->gang_leader = p->jobs[p->gang_leader_idx];
297
298 if (p->ctx->generation != p->gang_leader->generation) {
299 ret = -ECANCELED;
300 goto free_all_kdata;
301 }
302
303 if (p->uf_bo)
304 p->gang_leader->uf_addr = uf_offset;
305 kvfree(chunk_array);
306
307 /* Use this opportunity to fill in task info for the vm */
308 amdgpu_vm_set_task_info(vm);
309
310 return 0;
311
312free_all_kdata:
313 i = p->nchunks - 1;
314free_partial_kdata:
315 for (; i >= 0; i--)
316 kvfree(p->chunks[i].kdata);
317 kvfree(p->chunks);
318 p->chunks = NULL;
319 p->nchunks = 0;
320free_chunk:
321 kvfree(chunk_array);
322
323 return ret;
324}
325
326static int amdgpu_cs_p2_ib(struct amdgpu_cs_parser *p,
327 struct amdgpu_cs_chunk *chunk,
328 unsigned int *ce_preempt,
329 unsigned int *de_preempt)
330{
331 struct drm_amdgpu_cs_chunk_ib *chunk_ib = chunk->kdata;
332 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
333 struct amdgpu_vm *vm = &fpriv->vm;
334 struct amdgpu_ring *ring;
335 struct amdgpu_job *job;
336 struct amdgpu_ib *ib;
337 int r;
338
339 r = amdgpu_cs_job_idx(p, chunk_ib);
340 if (r < 0)
341 return r;
342
343 job = p->jobs[r];
344 ring = amdgpu_job_ring(job);
345 ib = &job->ibs[job->num_ibs++];
346
347 /* MM engine doesn't support user fences */
348 if (p->uf_bo && ring->funcs->no_user_fence)
349 return -EINVAL;
350
351 if (chunk_ib->ip_type == AMDGPU_HW_IP_GFX &&
352 chunk_ib->flags & AMDGPU_IB_FLAG_PREEMPT) {
353 if (chunk_ib->flags & AMDGPU_IB_FLAG_CE)
354 (*ce_preempt)++;
355 else
356 (*de_preempt)++;
357
358 /* Each GFX command submit allows only 1 IB max
359 * preemptible for CE & DE */
360 if (*ce_preempt > 1 || *de_preempt > 1)
361 return -EINVAL;
362 }
363
364 if (chunk_ib->flags & AMDGPU_IB_FLAG_PREAMBLE)
365 job->preamble_status |= AMDGPU_PREAMBLE_IB_PRESENT;
366
367 r = amdgpu_ib_get(p->adev, vm, ring->funcs->parse_cs ?
368 chunk_ib->ib_bytes : 0,
369 AMDGPU_IB_POOL_DELAYED, ib);
370 if (r) {
371 DRM_ERROR("Failed to get ib !\n");
372 return r;
373 }
374
375 ib->gpu_addr = chunk_ib->va_start;
376 ib->length_dw = chunk_ib->ib_bytes / 4;
377 ib->flags = chunk_ib->flags;
378 return 0;
379}
380
381static int amdgpu_cs_p2_dependencies(struct amdgpu_cs_parser *p,
382 struct amdgpu_cs_chunk *chunk)
383{
384 struct drm_amdgpu_cs_chunk_dep *deps = chunk->kdata;
385 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
386 unsigned int num_deps;
387 int i, r;
388
389 num_deps = chunk->length_dw * 4 /
390 sizeof(struct drm_amdgpu_cs_chunk_dep);
391
392 for (i = 0; i < num_deps; ++i) {
393 struct amdgpu_ctx *ctx;
394 struct drm_sched_entity *entity;
395 struct dma_fence *fence;
396
397 ctx = amdgpu_ctx_get(fpriv, deps[i].ctx_id);
398 if (ctx == NULL)
399 return -EINVAL;
400
401 r = amdgpu_ctx_get_entity(ctx, deps[i].ip_type,
402 deps[i].ip_instance,
403 deps[i].ring, &entity);
404 if (r) {
405 amdgpu_ctx_put(ctx);
406 return r;
407 }
408
409 fence = amdgpu_ctx_get_fence(ctx, entity, deps[i].handle);
410 amdgpu_ctx_put(ctx);
411
412 if (IS_ERR(fence))
413 return PTR_ERR(fence);
414 else if (!fence)
415 continue;
416
417 if (chunk->chunk_id == AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES) {
418 struct drm_sched_fence *s_fence;
419 struct dma_fence *old = fence;
420
421 s_fence = to_drm_sched_fence(fence);
422 fence = dma_fence_get(&s_fence->scheduled);
423 dma_fence_put(old);
424 }
425
426 r = amdgpu_sync_fence(&p->sync, fence);
427 dma_fence_put(fence);
428 if (r)
429 return r;
430 }
431 return 0;
432}
433
434static int amdgpu_syncobj_lookup_and_add(struct amdgpu_cs_parser *p,
435 uint32_t handle, u64 point,
436 u64 flags)
437{
438 struct dma_fence *fence;
439 int r;
440
441 r = drm_syncobj_find_fence(p->filp, handle, point, flags, &fence);
442 if (r) {
443 DRM_ERROR("syncobj %u failed to find fence @ %llu (%d)!\n",
444 handle, point, r);
445 return r;
446 }
447
448 r = amdgpu_sync_fence(&p->sync, fence);
449 dma_fence_put(fence);
450 return r;
451}
452
453static int amdgpu_cs_p2_syncobj_in(struct amdgpu_cs_parser *p,
454 struct amdgpu_cs_chunk *chunk)
455{
456 struct drm_amdgpu_cs_chunk_sem *deps = chunk->kdata;
457 unsigned int num_deps;
458 int i, r;
459
460 num_deps = chunk->length_dw * 4 /
461 sizeof(struct drm_amdgpu_cs_chunk_sem);
462 for (i = 0; i < num_deps; ++i) {
463 r = amdgpu_syncobj_lookup_and_add(p, deps[i].handle, 0, 0);
464 if (r)
465 return r;
466 }
467
468 return 0;
469}
470
471static int amdgpu_cs_p2_syncobj_timeline_wait(struct amdgpu_cs_parser *p,
472 struct amdgpu_cs_chunk *chunk)
473{
474 struct drm_amdgpu_cs_chunk_syncobj *syncobj_deps = chunk->kdata;
475 unsigned int num_deps;
476 int i, r;
477
478 num_deps = chunk->length_dw * 4 /
479 sizeof(struct drm_amdgpu_cs_chunk_syncobj);
480 for (i = 0; i < num_deps; ++i) {
481 r = amdgpu_syncobj_lookup_and_add(p, syncobj_deps[i].handle,
482 syncobj_deps[i].point,
483 syncobj_deps[i].flags);
484 if (r)
485 return r;
486 }
487
488 return 0;
489}
490
491static int amdgpu_cs_p2_syncobj_out(struct amdgpu_cs_parser *p,
492 struct amdgpu_cs_chunk *chunk)
493{
494 struct drm_amdgpu_cs_chunk_sem *deps = chunk->kdata;
495 unsigned int num_deps;
496 int i;
497
498 num_deps = chunk->length_dw * 4 /
499 sizeof(struct drm_amdgpu_cs_chunk_sem);
500
501 if (p->post_deps)
502 return -EINVAL;
503
504 p->post_deps = kmalloc_array(num_deps, sizeof(*p->post_deps),
505 GFP_KERNEL);
506 p->num_post_deps = 0;
507
508 if (!p->post_deps)
509 return -ENOMEM;
510
511
512 for (i = 0; i < num_deps; ++i) {
513 p->post_deps[i].syncobj =
514 drm_syncobj_find(p->filp, deps[i].handle);
515 if (!p->post_deps[i].syncobj)
516 return -EINVAL;
517 p->post_deps[i].chain = NULL;
518 p->post_deps[i].point = 0;
519 p->num_post_deps++;
520 }
521
522 return 0;
523}
524
525static int amdgpu_cs_p2_syncobj_timeline_signal(struct amdgpu_cs_parser *p,
526 struct amdgpu_cs_chunk *chunk)
527{
528 struct drm_amdgpu_cs_chunk_syncobj *syncobj_deps = chunk->kdata;
529 unsigned int num_deps;
530 int i;
531
532 num_deps = chunk->length_dw * 4 /
533 sizeof(struct drm_amdgpu_cs_chunk_syncobj);
534
535 if (p->post_deps)
536 return -EINVAL;
537
538 p->post_deps = kmalloc_array(num_deps, sizeof(*p->post_deps),
539 GFP_KERNEL);
540 p->num_post_deps = 0;
541
542 if (!p->post_deps)
543 return -ENOMEM;
544
545 for (i = 0; i < num_deps; ++i) {
546 struct amdgpu_cs_post_dep *dep = &p->post_deps[i];
547
548 dep->chain = NULL;
549 if (syncobj_deps[i].point) {
550 dep->chain = dma_fence_chain_alloc();
551 if (!dep->chain)
552 return -ENOMEM;
553 }
554
555 dep->syncobj = drm_syncobj_find(p->filp,
556 syncobj_deps[i].handle);
557 if (!dep->syncobj) {
558 dma_fence_chain_free(dep->chain);
559 return -EINVAL;
560 }
561 dep->point = syncobj_deps[i].point;
562 p->num_post_deps++;
563 }
564
565 return 0;
566}
567
568static int amdgpu_cs_p2_shadow(struct amdgpu_cs_parser *p,
569 struct amdgpu_cs_chunk *chunk)
570{
571 struct drm_amdgpu_cs_chunk_cp_gfx_shadow *shadow = chunk->kdata;
572 int i;
573
574 if (shadow->flags & ~AMDGPU_CS_CHUNK_CP_GFX_SHADOW_FLAGS_INIT_SHADOW)
575 return -EINVAL;
576
577 for (i = 0; i < p->gang_size; ++i) {
578 p->jobs[i]->shadow_va = shadow->shadow_va;
579 p->jobs[i]->csa_va = shadow->csa_va;
580 p->jobs[i]->gds_va = shadow->gds_va;
581 p->jobs[i]->init_shadow =
582 shadow->flags & AMDGPU_CS_CHUNK_CP_GFX_SHADOW_FLAGS_INIT_SHADOW;
583 }
584
585 return 0;
586}
587
588static int amdgpu_cs_pass2(struct amdgpu_cs_parser *p)
589{
590 unsigned int ce_preempt = 0, de_preempt = 0;
591 int i, r;
592
593 for (i = 0; i < p->nchunks; ++i) {
594 struct amdgpu_cs_chunk *chunk;
595
596 chunk = &p->chunks[i];
597
598 switch (chunk->chunk_id) {
599 case AMDGPU_CHUNK_ID_IB:
600 r = amdgpu_cs_p2_ib(p, chunk, &ce_preempt, &de_preempt);
601 if (r)
602 return r;
603 break;
604 case AMDGPU_CHUNK_ID_DEPENDENCIES:
605 case AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES:
606 r = amdgpu_cs_p2_dependencies(p, chunk);
607 if (r)
608 return r;
609 break;
610 case AMDGPU_CHUNK_ID_SYNCOBJ_IN:
611 r = amdgpu_cs_p2_syncobj_in(p, chunk);
612 if (r)
613 return r;
614 break;
615 case AMDGPU_CHUNK_ID_SYNCOBJ_OUT:
616 r = amdgpu_cs_p2_syncobj_out(p, chunk);
617 if (r)
618 return r;
619 break;
620 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_WAIT:
621 r = amdgpu_cs_p2_syncobj_timeline_wait(p, chunk);
622 if (r)
623 return r;
624 break;
625 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_SIGNAL:
626 r = amdgpu_cs_p2_syncobj_timeline_signal(p, chunk);
627 if (r)
628 return r;
629 break;
630 case AMDGPU_CHUNK_ID_CP_GFX_SHADOW:
631 r = amdgpu_cs_p2_shadow(p, chunk);
632 if (r)
633 return r;
634 break;
635 }
636 }
637
638 return 0;
639}
640
641/* Convert microseconds to bytes. */
642static u64 us_to_bytes(struct amdgpu_device *adev, s64 us)
643{
644 if (us <= 0 || !adev->mm_stats.log2_max_MBps)
645 return 0;
646
647 /* Since accum_us is incremented by a million per second, just
648 * multiply it by the number of MB/s to get the number of bytes.
649 */
650 return us << adev->mm_stats.log2_max_MBps;
651}
652
653static s64 bytes_to_us(struct amdgpu_device *adev, u64 bytes)
654{
655 if (!adev->mm_stats.log2_max_MBps)
656 return 0;
657
658 return bytes >> adev->mm_stats.log2_max_MBps;
659}
660
661/* Returns how many bytes TTM can move right now. If no bytes can be moved,
662 * it returns 0. If it returns non-zero, it's OK to move at least one buffer,
663 * which means it can go over the threshold once. If that happens, the driver
664 * will be in debt and no other buffer migrations can be done until that debt
665 * is repaid.
666 *
667 * This approach allows moving a buffer of any size (it's important to allow
668 * that).
669 *
670 * The currency is simply time in microseconds and it increases as the clock
671 * ticks. The accumulated microseconds (us) are converted to bytes and
672 * returned.
673 */
674static void amdgpu_cs_get_threshold_for_moves(struct amdgpu_device *adev,
675 u64 *max_bytes,
676 u64 *max_vis_bytes)
677{
678 s64 time_us, increment_us;
679 u64 free_vram, total_vram, used_vram;
680 /* Allow a maximum of 200 accumulated ms. This is basically per-IB
681 * throttling.
682 *
683 * It means that in order to get full max MBps, at least 5 IBs per
684 * second must be submitted and not more than 200ms apart from each
685 * other.
686 */
687 const s64 us_upper_bound = 200000;
688
689 if (!adev->mm_stats.log2_max_MBps) {
690 *max_bytes = 0;
691 *max_vis_bytes = 0;
692 return;
693 }
694
695 total_vram = adev->gmc.real_vram_size - atomic64_read(&adev->vram_pin_size);
696 used_vram = ttm_resource_manager_usage(&adev->mman.vram_mgr.manager);
697 free_vram = used_vram >= total_vram ? 0 : total_vram - used_vram;
698
699 spin_lock(&adev->mm_stats.lock);
700
701 /* Increase the amount of accumulated us. */
702 time_us = ktime_to_us(ktime_get());
703 increment_us = time_us - adev->mm_stats.last_update_us;
704 adev->mm_stats.last_update_us = time_us;
705 adev->mm_stats.accum_us = min(adev->mm_stats.accum_us + increment_us,
706 us_upper_bound);
707
708 /* This prevents the short period of low performance when the VRAM
709 * usage is low and the driver is in debt or doesn't have enough
710 * accumulated us to fill VRAM quickly.
711 *
712 * The situation can occur in these cases:
713 * - a lot of VRAM is freed by userspace
714 * - the presence of a big buffer causes a lot of evictions
715 * (solution: split buffers into smaller ones)
716 *
717 * If 128 MB or 1/8th of VRAM is free, start filling it now by setting
718 * accum_us to a positive number.
719 */
720 if (free_vram >= 128 * 1024 * 1024 || free_vram >= total_vram / 8) {
721 s64 min_us;
722
723 /* Be more aggressive on dGPUs. Try to fill a portion of free
724 * VRAM now.
725 */
726 if (!(adev->flags & AMD_IS_APU))
727 min_us = bytes_to_us(adev, free_vram / 4);
728 else
729 min_us = 0; /* Reset accum_us on APUs. */
730
731 adev->mm_stats.accum_us = max(min_us, adev->mm_stats.accum_us);
732 }
733
734 /* This is set to 0 if the driver is in debt to disallow (optional)
735 * buffer moves.
736 */
737 *max_bytes = us_to_bytes(adev, adev->mm_stats.accum_us);
738
739 /* Do the same for visible VRAM if half of it is free */
740 if (!amdgpu_gmc_vram_full_visible(&adev->gmc)) {
741 u64 total_vis_vram = adev->gmc.visible_vram_size;
742 u64 used_vis_vram =
743 amdgpu_vram_mgr_vis_usage(&adev->mman.vram_mgr);
744
745 if (used_vis_vram < total_vis_vram) {
746 u64 free_vis_vram = total_vis_vram - used_vis_vram;
747
748 adev->mm_stats.accum_us_vis = min(adev->mm_stats.accum_us_vis +
749 increment_us, us_upper_bound);
750
751 if (free_vis_vram >= total_vis_vram / 2)
752 adev->mm_stats.accum_us_vis =
753 max(bytes_to_us(adev, free_vis_vram / 2),
754 adev->mm_stats.accum_us_vis);
755 }
756
757 *max_vis_bytes = us_to_bytes(adev, adev->mm_stats.accum_us_vis);
758 } else {
759 *max_vis_bytes = 0;
760 }
761
762 spin_unlock(&adev->mm_stats.lock);
763}
764
765/* Report how many bytes have really been moved for the last command
766 * submission. This can result in a debt that can stop buffer migrations
767 * temporarily.
768 */
769void amdgpu_cs_report_moved_bytes(struct amdgpu_device *adev, u64 num_bytes,
770 u64 num_vis_bytes)
771{
772 spin_lock(&adev->mm_stats.lock);
773 adev->mm_stats.accum_us -= bytes_to_us(adev, num_bytes);
774 adev->mm_stats.accum_us_vis -= bytes_to_us(adev, num_vis_bytes);
775 spin_unlock(&adev->mm_stats.lock);
776}
777
778static int amdgpu_cs_bo_validate(void *param, struct amdgpu_bo *bo)
779{
780 struct amdgpu_device *adev = amdgpu_ttm_adev(bo->tbo.bdev);
781 struct amdgpu_cs_parser *p = param;
782 struct ttm_operation_ctx ctx = {
783 .interruptible = true,
784 .no_wait_gpu = false,
785 .resv = bo->tbo.base.resv
786 };
787 uint32_t domain;
788 int r;
789
790 if (bo->tbo.pin_count)
791 return 0;
792
793 /* Don't move this buffer if we have depleted our allowance
794 * to move it. Don't move anything if the threshold is zero.
795 */
796 if (p->bytes_moved < p->bytes_moved_threshold &&
797 (!bo->tbo.base.dma_buf ||
798 list_empty(&bo->tbo.base.dma_buf->attachments))) {
799 if (!amdgpu_gmc_vram_full_visible(&adev->gmc) &&
800 (bo->flags & AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED)) {
801 /* And don't move a CPU_ACCESS_REQUIRED BO to limited
802 * visible VRAM if we've depleted our allowance to do
803 * that.
804 */
805 if (p->bytes_moved_vis < p->bytes_moved_vis_threshold)
806 domain = bo->preferred_domains;
807 else
808 domain = bo->allowed_domains;
809 } else {
810 domain = bo->preferred_domains;
811 }
812 } else {
813 domain = bo->allowed_domains;
814 }
815
816retry:
817 amdgpu_bo_placement_from_domain(bo, domain);
818 r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
819
820 p->bytes_moved += ctx.bytes_moved;
821 if (!amdgpu_gmc_vram_full_visible(&adev->gmc) &&
822 amdgpu_res_cpu_visible(adev, bo->tbo.resource))
823 p->bytes_moved_vis += ctx.bytes_moved;
824
825 if (unlikely(r == -ENOMEM) && domain != bo->allowed_domains) {
826 domain = bo->allowed_domains;
827 goto retry;
828 }
829
830 return r;
831}
832
833static int amdgpu_cs_parser_bos(struct amdgpu_cs_parser *p,
834 union drm_amdgpu_cs *cs)
835{
836 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
837 struct ttm_operation_ctx ctx = { true, false };
838 struct amdgpu_vm *vm = &fpriv->vm;
839 struct amdgpu_bo_list_entry *e;
840 struct drm_gem_object *obj;
841 unsigned long index;
842 unsigned int i;
843 int r;
844
845 /* p->bo_list could already be assigned if AMDGPU_CHUNK_ID_BO_HANDLES is present */
846 if (cs->in.bo_list_handle) {
847 if (p->bo_list)
848 return -EINVAL;
849
850 r = amdgpu_bo_list_get(fpriv, cs->in.bo_list_handle,
851 &p->bo_list);
852 if (r)
853 return r;
854 } else if (!p->bo_list) {
855 /* Create a empty bo_list when no handle is provided */
856 r = amdgpu_bo_list_create(p->adev, p->filp, NULL, 0,
857 &p->bo_list);
858 if (r)
859 return r;
860 }
861
862 mutex_lock(&p->bo_list->bo_list_mutex);
863
864 /* Get userptr backing pages. If pages are updated after registered
865 * in amdgpu_gem_userptr_ioctl(), amdgpu_cs_list_validate() will do
866 * amdgpu_ttm_backend_bind() to flush and invalidate new pages
867 */
868 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
869 bool userpage_invalidated = false;
870 struct amdgpu_bo *bo = e->bo;
871 int i;
872
873 e->user_pages = kvcalloc(bo->tbo.ttm->num_pages,
874 sizeof(struct page *),
875 GFP_KERNEL);
876 if (!e->user_pages) {
877 DRM_ERROR("kvmalloc_array failure\n");
878 r = -ENOMEM;
879 goto out_free_user_pages;
880 }
881
882 r = amdgpu_ttm_tt_get_user_pages(bo, e->user_pages, &e->range);
883 if (r) {
884 kvfree(e->user_pages);
885 e->user_pages = NULL;
886 goto out_free_user_pages;
887 }
888
889 for (i = 0; i < bo->tbo.ttm->num_pages; i++) {
890 if (bo->tbo.ttm->pages[i] != e->user_pages[i]) {
891 userpage_invalidated = true;
892 break;
893 }
894 }
895 e->user_invalidated = userpage_invalidated;
896 }
897
898 drm_exec_until_all_locked(&p->exec) {
899 r = amdgpu_vm_lock_pd(&fpriv->vm, &p->exec, 1 + p->gang_size);
900 drm_exec_retry_on_contention(&p->exec);
901 if (unlikely(r))
902 goto out_free_user_pages;
903
904 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
905 /* One fence for TTM and one for each CS job */
906 r = drm_exec_prepare_obj(&p->exec, &e->bo->tbo.base,
907 1 + p->gang_size);
908 drm_exec_retry_on_contention(&p->exec);
909 if (unlikely(r))
910 goto out_free_user_pages;
911
912 e->bo_va = amdgpu_vm_bo_find(vm, e->bo);
913 }
914
915 if (p->uf_bo) {
916 r = drm_exec_prepare_obj(&p->exec, &p->uf_bo->tbo.base,
917 1 + p->gang_size);
918 drm_exec_retry_on_contention(&p->exec);
919 if (unlikely(r))
920 goto out_free_user_pages;
921 }
922 }
923
924 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
925 struct mm_struct *usermm;
926
927 usermm = amdgpu_ttm_tt_get_usermm(e->bo->tbo.ttm);
928 if (usermm && usermm != current->mm) {
929 r = -EPERM;
930 goto out_free_user_pages;
931 }
932
933 if (amdgpu_ttm_tt_is_userptr(e->bo->tbo.ttm) &&
934 e->user_invalidated && e->user_pages) {
935 amdgpu_bo_placement_from_domain(e->bo,
936 AMDGPU_GEM_DOMAIN_CPU);
937 r = ttm_bo_validate(&e->bo->tbo, &e->bo->placement,
938 &ctx);
939 if (r)
940 goto out_free_user_pages;
941
942 amdgpu_ttm_tt_set_user_pages(e->bo->tbo.ttm,
943 e->user_pages);
944 }
945
946 kvfree(e->user_pages);
947 e->user_pages = NULL;
948 }
949
950 amdgpu_cs_get_threshold_for_moves(p->adev, &p->bytes_moved_threshold,
951 &p->bytes_moved_vis_threshold);
952 p->bytes_moved = 0;
953 p->bytes_moved_vis = 0;
954
955 r = amdgpu_vm_validate(p->adev, &fpriv->vm, NULL,
956 amdgpu_cs_bo_validate, p);
957 if (r) {
958 DRM_ERROR("amdgpu_vm_validate() failed.\n");
959 goto out_free_user_pages;
960 }
961
962 drm_exec_for_each_locked_object(&p->exec, index, obj) {
963 r = amdgpu_cs_bo_validate(p, gem_to_amdgpu_bo(obj));
964 if (unlikely(r))
965 goto out_free_user_pages;
966 }
967
968 if (p->uf_bo) {
969 r = amdgpu_ttm_alloc_gart(&p->uf_bo->tbo);
970 if (unlikely(r))
971 goto out_free_user_pages;
972
973 p->gang_leader->uf_addr += amdgpu_bo_gpu_offset(p->uf_bo);
974 }
975
976 amdgpu_cs_report_moved_bytes(p->adev, p->bytes_moved,
977 p->bytes_moved_vis);
978
979 for (i = 0; i < p->gang_size; ++i)
980 amdgpu_job_set_resources(p->jobs[i], p->bo_list->gds_obj,
981 p->bo_list->gws_obj,
982 p->bo_list->oa_obj);
983 return 0;
984
985out_free_user_pages:
986 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
987 struct amdgpu_bo *bo = e->bo;
988
989 if (!e->user_pages)
990 continue;
991 amdgpu_ttm_tt_get_user_pages_done(bo->tbo.ttm, e->range);
992 kvfree(e->user_pages);
993 e->user_pages = NULL;
994 e->range = NULL;
995 }
996 mutex_unlock(&p->bo_list->bo_list_mutex);
997 return r;
998}
999
1000static void trace_amdgpu_cs_ibs(struct amdgpu_cs_parser *p)
1001{
1002 int i, j;
1003
1004 if (!trace_amdgpu_cs_enabled())
1005 return;
1006
1007 for (i = 0; i < p->gang_size; ++i) {
1008 struct amdgpu_job *job = p->jobs[i];
1009
1010 for (j = 0; j < job->num_ibs; ++j)
1011 trace_amdgpu_cs(p, job, &job->ibs[j]);
1012 }
1013}
1014
1015static int amdgpu_cs_patch_ibs(struct amdgpu_cs_parser *p,
1016 struct amdgpu_job *job)
1017{
1018 struct amdgpu_ring *ring = amdgpu_job_ring(job);
1019 unsigned int i;
1020 int r;
1021
1022 /* Only for UVD/VCE VM emulation */
1023 if (!ring->funcs->parse_cs && !ring->funcs->patch_cs_in_place)
1024 return 0;
1025
1026 for (i = 0; i < job->num_ibs; ++i) {
1027 struct amdgpu_ib *ib = &job->ibs[i];
1028 struct amdgpu_bo_va_mapping *m;
1029 struct amdgpu_bo *aobj;
1030 uint64_t va_start;
1031 uint8_t *kptr;
1032
1033 va_start = ib->gpu_addr & AMDGPU_GMC_HOLE_MASK;
1034 r = amdgpu_cs_find_mapping(p, va_start, &aobj, &m);
1035 if (r) {
1036 DRM_ERROR("IB va_start is invalid\n");
1037 return r;
1038 }
1039
1040 if ((va_start + ib->length_dw * 4) >
1041 (m->last + 1) * AMDGPU_GPU_PAGE_SIZE) {
1042 DRM_ERROR("IB va_start+ib_bytes is invalid\n");
1043 return -EINVAL;
1044 }
1045
1046 /* the IB should be reserved at this point */
1047 r = amdgpu_bo_kmap(aobj, (void **)&kptr);
1048 if (r)
1049 return r;
1050
1051 kptr += va_start - (m->start * AMDGPU_GPU_PAGE_SIZE);
1052
1053 if (ring->funcs->parse_cs) {
1054 memcpy(ib->ptr, kptr, ib->length_dw * 4);
1055 amdgpu_bo_kunmap(aobj);
1056
1057 r = amdgpu_ring_parse_cs(ring, p, job, ib);
1058 if (r)
1059 return r;
1060 } else {
1061 ib->ptr = (uint32_t *)kptr;
1062 r = amdgpu_ring_patch_cs_in_place(ring, p, job, ib);
1063 amdgpu_bo_kunmap(aobj);
1064 if (r)
1065 return r;
1066 }
1067 }
1068
1069 return 0;
1070}
1071
1072static int amdgpu_cs_patch_jobs(struct amdgpu_cs_parser *p)
1073{
1074 unsigned int i;
1075 int r;
1076
1077 for (i = 0; i < p->gang_size; ++i) {
1078 r = amdgpu_cs_patch_ibs(p, p->jobs[i]);
1079 if (r)
1080 return r;
1081 }
1082 return 0;
1083}
1084
1085static int amdgpu_cs_vm_handling(struct amdgpu_cs_parser *p)
1086{
1087 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
1088 struct amdgpu_job *job = p->gang_leader;
1089 struct amdgpu_device *adev = p->adev;
1090 struct amdgpu_vm *vm = &fpriv->vm;
1091 struct amdgpu_bo_list_entry *e;
1092 struct amdgpu_bo_va *bo_va;
1093 unsigned int i;
1094 int r;
1095
1096 r = amdgpu_vm_clear_freed(adev, vm, NULL);
1097 if (r)
1098 return r;
1099
1100 r = amdgpu_vm_bo_update(adev, fpriv->prt_va, false);
1101 if (r)
1102 return r;
1103
1104 r = amdgpu_sync_fence(&p->sync, fpriv->prt_va->last_pt_update);
1105 if (r)
1106 return r;
1107
1108 if (fpriv->csa_va) {
1109 bo_va = fpriv->csa_va;
1110 BUG_ON(!bo_va);
1111 r = amdgpu_vm_bo_update(adev, bo_va, false);
1112 if (r)
1113 return r;
1114
1115 r = amdgpu_sync_fence(&p->sync, bo_va->last_pt_update);
1116 if (r)
1117 return r;
1118 }
1119
1120 /* FIXME: In theory this loop shouldn't be needed any more when
1121 * amdgpu_vm_handle_moved handles all moved BOs that are reserved
1122 * with p->ticket. But removing it caused test regressions, so I'm
1123 * leaving it here for now.
1124 */
1125 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
1126 bo_va = e->bo_va;
1127 if (bo_va == NULL)
1128 continue;
1129
1130 r = amdgpu_vm_bo_update(adev, bo_va, false);
1131 if (r)
1132 return r;
1133
1134 r = amdgpu_sync_fence(&p->sync, bo_va->last_pt_update);
1135 if (r)
1136 return r;
1137 }
1138
1139 r = amdgpu_vm_handle_moved(adev, vm, &p->exec.ticket);
1140 if (r)
1141 return r;
1142
1143 r = amdgpu_vm_update_pdes(adev, vm, false);
1144 if (r)
1145 return r;
1146
1147 r = amdgpu_sync_fence(&p->sync, vm->last_update);
1148 if (r)
1149 return r;
1150
1151 for (i = 0; i < p->gang_size; ++i) {
1152 job = p->jobs[i];
1153
1154 if (!job->vm)
1155 continue;
1156
1157 job->vm_pd_addr = amdgpu_gmc_pd_addr(vm->root.bo);
1158 }
1159
1160 if (adev->debug_vm) {
1161 /* Invalidate all BOs to test for userspace bugs */
1162 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
1163 struct amdgpu_bo *bo = e->bo;
1164
1165 /* ignore duplicates */
1166 if (!bo)
1167 continue;
1168
1169 amdgpu_vm_bo_invalidate(adev, bo, false);
1170 }
1171 }
1172
1173 return 0;
1174}
1175
1176static int amdgpu_cs_sync_rings(struct amdgpu_cs_parser *p)
1177{
1178 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
1179 struct drm_gpu_scheduler *sched;
1180 struct drm_gem_object *obj;
1181 struct dma_fence *fence;
1182 unsigned long index;
1183 unsigned int i;
1184 int r;
1185
1186 r = amdgpu_ctx_wait_prev_fence(p->ctx, p->entities[p->gang_leader_idx]);
1187 if (r) {
1188 if (r != -ERESTARTSYS)
1189 DRM_ERROR("amdgpu_ctx_wait_prev_fence failed.\n");
1190 return r;
1191 }
1192
1193 drm_exec_for_each_locked_object(&p->exec, index, obj) {
1194 struct amdgpu_bo *bo = gem_to_amdgpu_bo(obj);
1195
1196 struct dma_resv *resv = bo->tbo.base.resv;
1197 enum amdgpu_sync_mode sync_mode;
1198
1199 sync_mode = amdgpu_bo_explicit_sync(bo) ?
1200 AMDGPU_SYNC_EXPLICIT : AMDGPU_SYNC_NE_OWNER;
1201 r = amdgpu_sync_resv(p->adev, &p->sync, resv, sync_mode,
1202 &fpriv->vm);
1203 if (r)
1204 return r;
1205 }
1206
1207 for (i = 0; i < p->gang_size; ++i) {
1208 r = amdgpu_sync_push_to_job(&p->sync, p->jobs[i]);
1209 if (r)
1210 return r;
1211 }
1212
1213 sched = p->gang_leader->base.entity->rq->sched;
1214 while ((fence = amdgpu_sync_get_fence(&p->sync))) {
1215 struct drm_sched_fence *s_fence = to_drm_sched_fence(fence);
1216
1217 /*
1218 * When we have an dependency it might be necessary to insert a
1219 * pipeline sync to make sure that all caches etc are flushed and the
1220 * next job actually sees the results from the previous one
1221 * before we start executing on the same scheduler ring.
1222 */
1223 if (!s_fence || s_fence->sched != sched) {
1224 dma_fence_put(fence);
1225 continue;
1226 }
1227
1228 r = amdgpu_sync_fence(&p->gang_leader->explicit_sync, fence);
1229 dma_fence_put(fence);
1230 if (r)
1231 return r;
1232 }
1233 return 0;
1234}
1235
1236static void amdgpu_cs_post_dependencies(struct amdgpu_cs_parser *p)
1237{
1238 int i;
1239
1240 for (i = 0; i < p->num_post_deps; ++i) {
1241 if (p->post_deps[i].chain && p->post_deps[i].point) {
1242 drm_syncobj_add_point(p->post_deps[i].syncobj,
1243 p->post_deps[i].chain,
1244 p->fence, p->post_deps[i].point);
1245 p->post_deps[i].chain = NULL;
1246 } else {
1247 drm_syncobj_replace_fence(p->post_deps[i].syncobj,
1248 p->fence);
1249 }
1250 }
1251}
1252
1253static int amdgpu_cs_submit(struct amdgpu_cs_parser *p,
1254 union drm_amdgpu_cs *cs)
1255{
1256 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
1257 struct amdgpu_job *leader = p->gang_leader;
1258 struct amdgpu_bo_list_entry *e;
1259 struct drm_gem_object *gobj;
1260 unsigned long index;
1261 unsigned int i;
1262 uint64_t seq;
1263 int r;
1264
1265 for (i = 0; i < p->gang_size; ++i)
1266 drm_sched_job_arm(&p->jobs[i]->base);
1267
1268 for (i = 0; i < p->gang_size; ++i) {
1269 struct dma_fence *fence;
1270
1271 if (p->jobs[i] == leader)
1272 continue;
1273
1274 fence = &p->jobs[i]->base.s_fence->scheduled;
1275 dma_fence_get(fence);
1276 r = drm_sched_job_add_dependency(&leader->base, fence);
1277 if (r) {
1278 dma_fence_put(fence);
1279 return r;
1280 }
1281 }
1282
1283 if (p->gang_size > 1) {
1284 for (i = 0; i < p->gang_size; ++i)
1285 amdgpu_job_set_gang_leader(p->jobs[i], leader);
1286 }
1287
1288 /* No memory allocation is allowed while holding the notifier lock.
1289 * The lock is held until amdgpu_cs_submit is finished and fence is
1290 * added to BOs.
1291 */
1292 mutex_lock(&p->adev->notifier_lock);
1293
1294 /* If userptr are invalidated after amdgpu_cs_parser_bos(), return
1295 * -EAGAIN, drmIoctl in libdrm will restart the amdgpu_cs_ioctl.
1296 */
1297 r = 0;
1298 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
1299 r |= !amdgpu_ttm_tt_get_user_pages_done(e->bo->tbo.ttm,
1300 e->range);
1301 e->range = NULL;
1302 }
1303 if (r) {
1304 r = -EAGAIN;
1305 mutex_unlock(&p->adev->notifier_lock);
1306 return r;
1307 }
1308
1309 p->fence = dma_fence_get(&leader->base.s_fence->finished);
1310 drm_exec_for_each_locked_object(&p->exec, index, gobj) {
1311
1312 ttm_bo_move_to_lru_tail_unlocked(&gem_to_amdgpu_bo(gobj)->tbo);
1313
1314 /* Everybody except for the gang leader uses READ */
1315 for (i = 0; i < p->gang_size; ++i) {
1316 if (p->jobs[i] == leader)
1317 continue;
1318
1319 dma_resv_add_fence(gobj->resv,
1320 &p->jobs[i]->base.s_fence->finished,
1321 DMA_RESV_USAGE_READ);
1322 }
1323
1324 /* The gang leader as remembered as writer */
1325 dma_resv_add_fence(gobj->resv, p->fence, DMA_RESV_USAGE_WRITE);
1326 }
1327
1328 seq = amdgpu_ctx_add_fence(p->ctx, p->entities[p->gang_leader_idx],
1329 p->fence);
1330 amdgpu_cs_post_dependencies(p);
1331
1332 if ((leader->preamble_status & AMDGPU_PREAMBLE_IB_PRESENT) &&
1333 !p->ctx->preamble_presented) {
1334 leader->preamble_status |= AMDGPU_PREAMBLE_IB_PRESENT_FIRST;
1335 p->ctx->preamble_presented = true;
1336 }
1337
1338 cs->out.handle = seq;
1339 leader->uf_sequence = seq;
1340
1341 amdgpu_vm_bo_trace_cs(&fpriv->vm, &p->exec.ticket);
1342 for (i = 0; i < p->gang_size; ++i) {
1343 amdgpu_job_free_resources(p->jobs[i]);
1344 trace_amdgpu_cs_ioctl(p->jobs[i]);
1345 drm_sched_entity_push_job(&p->jobs[i]->base);
1346 p->jobs[i] = NULL;
1347 }
1348
1349 amdgpu_vm_move_to_lru_tail(p->adev, &fpriv->vm);
1350
1351 mutex_unlock(&p->adev->notifier_lock);
1352 mutex_unlock(&p->bo_list->bo_list_mutex);
1353 return 0;
1354}
1355
1356/* Cleanup the parser structure */
1357static void amdgpu_cs_parser_fini(struct amdgpu_cs_parser *parser)
1358{
1359 unsigned int i;
1360
1361 amdgpu_sync_free(&parser->sync);
1362 drm_exec_fini(&parser->exec);
1363
1364 for (i = 0; i < parser->num_post_deps; i++) {
1365 drm_syncobj_put(parser->post_deps[i].syncobj);
1366 kfree(parser->post_deps[i].chain);
1367 }
1368 kfree(parser->post_deps);
1369
1370 dma_fence_put(parser->fence);
1371
1372 if (parser->ctx)
1373 amdgpu_ctx_put(parser->ctx);
1374 if (parser->bo_list)
1375 amdgpu_bo_list_put(parser->bo_list);
1376
1377 for (i = 0; i < parser->nchunks; i++)
1378 kvfree(parser->chunks[i].kdata);
1379 kvfree(parser->chunks);
1380 for (i = 0; i < parser->gang_size; ++i) {
1381 if (parser->jobs[i])
1382 amdgpu_job_free(parser->jobs[i]);
1383 }
1384 amdgpu_bo_unref(&parser->uf_bo);
1385}
1386
1387int amdgpu_cs_ioctl(struct drm_device *dev, void *data, struct drm_file *filp)
1388{
1389 struct amdgpu_device *adev = drm_to_adev(dev);
1390 struct amdgpu_cs_parser parser;
1391 int r;
1392
1393 if (amdgpu_ras_intr_triggered())
1394 return -EHWPOISON;
1395
1396 if (!adev->accel_working)
1397 return -EBUSY;
1398
1399 r = amdgpu_cs_parser_init(&parser, adev, filp, data);
1400 if (r) {
1401 DRM_ERROR_RATELIMITED("Failed to initialize parser %d!\n", r);
1402 return r;
1403 }
1404
1405 r = amdgpu_cs_pass1(&parser, data);
1406 if (r)
1407 goto error_fini;
1408
1409 r = amdgpu_cs_pass2(&parser);
1410 if (r)
1411 goto error_fini;
1412
1413 r = amdgpu_cs_parser_bos(&parser, data);
1414 if (r) {
1415 if (r == -ENOMEM)
1416 DRM_ERROR("Not enough memory for command submission!\n");
1417 else if (r != -ERESTARTSYS && r != -EAGAIN)
1418 DRM_DEBUG("Failed to process the buffer list %d!\n", r);
1419 goto error_fini;
1420 }
1421
1422 r = amdgpu_cs_patch_jobs(&parser);
1423 if (r)
1424 goto error_backoff;
1425
1426 r = amdgpu_cs_vm_handling(&parser);
1427 if (r)
1428 goto error_backoff;
1429
1430 r = amdgpu_cs_sync_rings(&parser);
1431 if (r)
1432 goto error_backoff;
1433
1434 trace_amdgpu_cs_ibs(&parser);
1435
1436 r = amdgpu_cs_submit(&parser, data);
1437 if (r)
1438 goto error_backoff;
1439
1440 amdgpu_cs_parser_fini(&parser);
1441 return 0;
1442
1443error_backoff:
1444 mutex_unlock(&parser.bo_list->bo_list_mutex);
1445
1446error_fini:
1447 amdgpu_cs_parser_fini(&parser);
1448 return r;
1449}
1450
1451/**
1452 * amdgpu_cs_wait_ioctl - wait for a command submission to finish
1453 *
1454 * @dev: drm device
1455 * @data: data from userspace
1456 * @filp: file private
1457 *
1458 * Wait for the command submission identified by handle to finish.
1459 */
1460int amdgpu_cs_wait_ioctl(struct drm_device *dev, void *data,
1461 struct drm_file *filp)
1462{
1463 union drm_amdgpu_wait_cs *wait = data;
1464 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout);
1465 struct drm_sched_entity *entity;
1466 struct amdgpu_ctx *ctx;
1467 struct dma_fence *fence;
1468 long r;
1469
1470 ctx = amdgpu_ctx_get(filp->driver_priv, wait->in.ctx_id);
1471 if (ctx == NULL)
1472 return -EINVAL;
1473
1474 r = amdgpu_ctx_get_entity(ctx, wait->in.ip_type, wait->in.ip_instance,
1475 wait->in.ring, &entity);
1476 if (r) {
1477 amdgpu_ctx_put(ctx);
1478 return r;
1479 }
1480
1481 fence = amdgpu_ctx_get_fence(ctx, entity, wait->in.handle);
1482 if (IS_ERR(fence))
1483 r = PTR_ERR(fence);
1484 else if (fence) {
1485 r = dma_fence_wait_timeout(fence, true, timeout);
1486 if (r > 0 && fence->error)
1487 r = fence->error;
1488 dma_fence_put(fence);
1489 } else
1490 r = 1;
1491
1492 amdgpu_ctx_put(ctx);
1493 if (r < 0)
1494 return r;
1495
1496 memset(wait, 0, sizeof(*wait));
1497 wait->out.status = (r == 0);
1498
1499 return 0;
1500}
1501
1502/**
1503 * amdgpu_cs_get_fence - helper to get fence from drm_amdgpu_fence
1504 *
1505 * @adev: amdgpu device
1506 * @filp: file private
1507 * @user: drm_amdgpu_fence copied from user space
1508 */
1509static struct dma_fence *amdgpu_cs_get_fence(struct amdgpu_device *adev,
1510 struct drm_file *filp,
1511 struct drm_amdgpu_fence *user)
1512{
1513 struct drm_sched_entity *entity;
1514 struct amdgpu_ctx *ctx;
1515 struct dma_fence *fence;
1516 int r;
1517
1518 ctx = amdgpu_ctx_get(filp->driver_priv, user->ctx_id);
1519 if (ctx == NULL)
1520 return ERR_PTR(-EINVAL);
1521
1522 r = amdgpu_ctx_get_entity(ctx, user->ip_type, user->ip_instance,
1523 user->ring, &entity);
1524 if (r) {
1525 amdgpu_ctx_put(ctx);
1526 return ERR_PTR(r);
1527 }
1528
1529 fence = amdgpu_ctx_get_fence(ctx, entity, user->seq_no);
1530 amdgpu_ctx_put(ctx);
1531
1532 return fence;
1533}
1534
1535int amdgpu_cs_fence_to_handle_ioctl(struct drm_device *dev, void *data,
1536 struct drm_file *filp)
1537{
1538 struct amdgpu_device *adev = drm_to_adev(dev);
1539 union drm_amdgpu_fence_to_handle *info = data;
1540 struct dma_fence *fence;
1541 struct drm_syncobj *syncobj;
1542 struct sync_file *sync_file;
1543 int fd, r;
1544
1545 fence = amdgpu_cs_get_fence(adev, filp, &info->in.fence);
1546 if (IS_ERR(fence))
1547 return PTR_ERR(fence);
1548
1549 if (!fence)
1550 fence = dma_fence_get_stub();
1551
1552 switch (info->in.what) {
1553 case AMDGPU_FENCE_TO_HANDLE_GET_SYNCOBJ:
1554 r = drm_syncobj_create(&syncobj, 0, fence);
1555 dma_fence_put(fence);
1556 if (r)
1557 return r;
1558 r = drm_syncobj_get_handle(filp, syncobj, &info->out.handle);
1559 drm_syncobj_put(syncobj);
1560 return r;
1561
1562 case AMDGPU_FENCE_TO_HANDLE_GET_SYNCOBJ_FD:
1563 r = drm_syncobj_create(&syncobj, 0, fence);
1564 dma_fence_put(fence);
1565 if (r)
1566 return r;
1567 r = drm_syncobj_get_fd(syncobj, (int *)&info->out.handle);
1568 drm_syncobj_put(syncobj);
1569 return r;
1570
1571 case AMDGPU_FENCE_TO_HANDLE_GET_SYNC_FILE_FD:
1572 fd = get_unused_fd_flags(O_CLOEXEC);
1573 if (fd < 0) {
1574 dma_fence_put(fence);
1575 return fd;
1576 }
1577
1578 sync_file = sync_file_create(fence);
1579 dma_fence_put(fence);
1580 if (!sync_file) {
1581 put_unused_fd(fd);
1582 return -ENOMEM;
1583 }
1584
1585 fd_install(fd, sync_file->file);
1586 info->out.handle = fd;
1587 return 0;
1588
1589 default:
1590 dma_fence_put(fence);
1591 return -EINVAL;
1592 }
1593}
1594
1595/**
1596 * amdgpu_cs_wait_all_fences - wait on all fences to signal
1597 *
1598 * @adev: amdgpu device
1599 * @filp: file private
1600 * @wait: wait parameters
1601 * @fences: array of drm_amdgpu_fence
1602 */
1603static int amdgpu_cs_wait_all_fences(struct amdgpu_device *adev,
1604 struct drm_file *filp,
1605 union drm_amdgpu_wait_fences *wait,
1606 struct drm_amdgpu_fence *fences)
1607{
1608 uint32_t fence_count = wait->in.fence_count;
1609 unsigned int i;
1610 long r = 1;
1611
1612 for (i = 0; i < fence_count; i++) {
1613 struct dma_fence *fence;
1614 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout_ns);
1615
1616 fence = amdgpu_cs_get_fence(adev, filp, &fences[i]);
1617 if (IS_ERR(fence))
1618 return PTR_ERR(fence);
1619 else if (!fence)
1620 continue;
1621
1622 r = dma_fence_wait_timeout(fence, true, timeout);
1623 if (r > 0 && fence->error)
1624 r = fence->error;
1625
1626 dma_fence_put(fence);
1627 if (r < 0)
1628 return r;
1629
1630 if (r == 0)
1631 break;
1632 }
1633
1634 memset(wait, 0, sizeof(*wait));
1635 wait->out.status = (r > 0);
1636
1637 return 0;
1638}
1639
1640/**
1641 * amdgpu_cs_wait_any_fence - wait on any fence to signal
1642 *
1643 * @adev: amdgpu device
1644 * @filp: file private
1645 * @wait: wait parameters
1646 * @fences: array of drm_amdgpu_fence
1647 */
1648static int amdgpu_cs_wait_any_fence(struct amdgpu_device *adev,
1649 struct drm_file *filp,
1650 union drm_amdgpu_wait_fences *wait,
1651 struct drm_amdgpu_fence *fences)
1652{
1653 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout_ns);
1654 uint32_t fence_count = wait->in.fence_count;
1655 uint32_t first = ~0;
1656 struct dma_fence **array;
1657 unsigned int i;
1658 long r;
1659
1660 /* Prepare the fence array */
1661 array = kcalloc(fence_count, sizeof(struct dma_fence *), GFP_KERNEL);
1662
1663 if (array == NULL)
1664 return -ENOMEM;
1665
1666 for (i = 0; i < fence_count; i++) {
1667 struct dma_fence *fence;
1668
1669 fence = amdgpu_cs_get_fence(adev, filp, &fences[i]);
1670 if (IS_ERR(fence)) {
1671 r = PTR_ERR(fence);
1672 goto err_free_fence_array;
1673 } else if (fence) {
1674 array[i] = fence;
1675 } else { /* NULL, the fence has been already signaled */
1676 r = 1;
1677 first = i;
1678 goto out;
1679 }
1680 }
1681
1682 r = dma_fence_wait_any_timeout(array, fence_count, true, timeout,
1683 &first);
1684 if (r < 0)
1685 goto err_free_fence_array;
1686
1687out:
1688 memset(wait, 0, sizeof(*wait));
1689 wait->out.status = (r > 0);
1690 wait->out.first_signaled = first;
1691
1692 if (first < fence_count && array[first])
1693 r = array[first]->error;
1694 else
1695 r = 0;
1696
1697err_free_fence_array:
1698 for (i = 0; i < fence_count; i++)
1699 dma_fence_put(array[i]);
1700 kfree(array);
1701
1702 return r;
1703}
1704
1705/**
1706 * amdgpu_cs_wait_fences_ioctl - wait for multiple command submissions to finish
1707 *
1708 * @dev: drm device
1709 * @data: data from userspace
1710 * @filp: file private
1711 */
1712int amdgpu_cs_wait_fences_ioctl(struct drm_device *dev, void *data,
1713 struct drm_file *filp)
1714{
1715 struct amdgpu_device *adev = drm_to_adev(dev);
1716 union drm_amdgpu_wait_fences *wait = data;
1717 uint32_t fence_count = wait->in.fence_count;
1718 struct drm_amdgpu_fence *fences_user;
1719 struct drm_amdgpu_fence *fences;
1720 int r;
1721
1722 /* Get the fences from userspace */
1723 fences = kmalloc_array(fence_count, sizeof(struct drm_amdgpu_fence),
1724 GFP_KERNEL);
1725 if (fences == NULL)
1726 return -ENOMEM;
1727
1728 fences_user = u64_to_user_ptr(wait->in.fences);
1729 if (copy_from_user(fences, fences_user,
1730 sizeof(struct drm_amdgpu_fence) * fence_count)) {
1731 r = -EFAULT;
1732 goto err_free_fences;
1733 }
1734
1735 if (wait->in.wait_all)
1736 r = amdgpu_cs_wait_all_fences(adev, filp, wait, fences);
1737 else
1738 r = amdgpu_cs_wait_any_fence(adev, filp, wait, fences);
1739
1740err_free_fences:
1741 kfree(fences);
1742
1743 return r;
1744}
1745
1746/**
1747 * amdgpu_cs_find_mapping - find bo_va for VM address
1748 *
1749 * @parser: command submission parser context
1750 * @addr: VM address
1751 * @bo: resulting BO of the mapping found
1752 * @map: Placeholder to return found BO mapping
1753 *
1754 * Search the buffer objects in the command submission context for a certain
1755 * virtual memory address. Returns allocation structure when found, NULL
1756 * otherwise.
1757 */
1758int amdgpu_cs_find_mapping(struct amdgpu_cs_parser *parser,
1759 uint64_t addr, struct amdgpu_bo **bo,
1760 struct amdgpu_bo_va_mapping **map)
1761{
1762 struct amdgpu_fpriv *fpriv = parser->filp->driver_priv;
1763 struct ttm_operation_ctx ctx = { false, false };
1764 struct amdgpu_vm *vm = &fpriv->vm;
1765 struct amdgpu_bo_va_mapping *mapping;
1766 int r;
1767
1768 addr /= AMDGPU_GPU_PAGE_SIZE;
1769
1770 mapping = amdgpu_vm_bo_lookup_mapping(vm, addr);
1771 if (!mapping || !mapping->bo_va || !mapping->bo_va->base.bo)
1772 return -EINVAL;
1773
1774 *bo = mapping->bo_va->base.bo;
1775 *map = mapping;
1776
1777 /* Double check that the BO is reserved by this CS */
1778 if (dma_resv_locking_ctx((*bo)->tbo.base.resv) != &parser->exec.ticket)
1779 return -EINVAL;
1780
1781 if (!((*bo)->flags & AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS)) {
1782 (*bo)->flags |= AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS;
1783 amdgpu_bo_placement_from_domain(*bo, (*bo)->allowed_domains);
1784 r = ttm_bo_validate(&(*bo)->tbo, &(*bo)->placement, &ctx);
1785 if (r)
1786 return r;
1787 }
1788
1789 return amdgpu_ttm_alloc_gart(&(*bo)->tbo);
1790}
1/*
2 * Copyright 2008 Jerome Glisse.
3 * All Rights Reserved.
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
20 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
21 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *
24 * Authors:
25 * Jerome Glisse <glisse@freedesktop.org>
26 */
27
28#include <linux/file.h>
29#include <linux/pagemap.h>
30#include <linux/sync_file.h>
31#include <linux/dma-buf.h>
32
33#include <drm/amdgpu_drm.h>
34#include <drm/drm_syncobj.h>
35#include "amdgpu.h"
36#include "amdgpu_trace.h"
37#include "amdgpu_gmc.h"
38#include "amdgpu_gem.h"
39#include "amdgpu_ras.h"
40
41static int amdgpu_cs_user_fence_chunk(struct amdgpu_cs_parser *p,
42 struct drm_amdgpu_cs_chunk_fence *data,
43 uint32_t *offset)
44{
45 struct drm_gem_object *gobj;
46 struct amdgpu_bo *bo;
47 unsigned long size;
48 int r;
49
50 gobj = drm_gem_object_lookup(p->filp, data->handle);
51 if (gobj == NULL)
52 return -EINVAL;
53
54 bo = amdgpu_bo_ref(gem_to_amdgpu_bo(gobj));
55 p->uf_entry.priority = 0;
56 p->uf_entry.tv.bo = &bo->tbo;
57 /* One for TTM and one for the CS job */
58 p->uf_entry.tv.num_shared = 2;
59
60 drm_gem_object_put(gobj);
61
62 size = amdgpu_bo_size(bo);
63 if (size != PAGE_SIZE || (data->offset + 8) > size) {
64 r = -EINVAL;
65 goto error_unref;
66 }
67
68 if (amdgpu_ttm_tt_get_usermm(bo->tbo.ttm)) {
69 r = -EINVAL;
70 goto error_unref;
71 }
72
73 *offset = data->offset;
74
75 return 0;
76
77error_unref:
78 amdgpu_bo_unref(&bo);
79 return r;
80}
81
82static int amdgpu_cs_bo_handles_chunk(struct amdgpu_cs_parser *p,
83 struct drm_amdgpu_bo_list_in *data)
84{
85 int r;
86 struct drm_amdgpu_bo_list_entry *info = NULL;
87
88 r = amdgpu_bo_create_list_entry_array(data, &info);
89 if (r)
90 return r;
91
92 r = amdgpu_bo_list_create(p->adev, p->filp, info, data->bo_number,
93 &p->bo_list);
94 if (r)
95 goto error_free;
96
97 kvfree(info);
98 return 0;
99
100error_free:
101 if (info)
102 kvfree(info);
103
104 return r;
105}
106
107static int amdgpu_cs_parser_init(struct amdgpu_cs_parser *p, union drm_amdgpu_cs *cs)
108{
109 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
110 struct amdgpu_vm *vm = &fpriv->vm;
111 uint64_t *chunk_array_user;
112 uint64_t *chunk_array;
113 unsigned size, num_ibs = 0;
114 uint32_t uf_offset = 0;
115 int i;
116 int ret;
117
118 if (cs->in.num_chunks == 0)
119 return 0;
120
121 chunk_array = kmalloc_array(cs->in.num_chunks, sizeof(uint64_t), GFP_KERNEL);
122 if (!chunk_array)
123 return -ENOMEM;
124
125 p->ctx = amdgpu_ctx_get(fpriv, cs->in.ctx_id);
126 if (!p->ctx) {
127 ret = -EINVAL;
128 goto free_chunk;
129 }
130
131 mutex_lock(&p->ctx->lock);
132
133 /* skip guilty context job */
134 if (atomic_read(&p->ctx->guilty) == 1) {
135 ret = -ECANCELED;
136 goto free_chunk;
137 }
138
139 /* get chunks */
140 chunk_array_user = u64_to_user_ptr(cs->in.chunks);
141 if (copy_from_user(chunk_array, chunk_array_user,
142 sizeof(uint64_t)*cs->in.num_chunks)) {
143 ret = -EFAULT;
144 goto free_chunk;
145 }
146
147 p->nchunks = cs->in.num_chunks;
148 p->chunks = kmalloc_array(p->nchunks, sizeof(struct amdgpu_cs_chunk),
149 GFP_KERNEL);
150 if (!p->chunks) {
151 ret = -ENOMEM;
152 goto free_chunk;
153 }
154
155 for (i = 0; i < p->nchunks; i++) {
156 struct drm_amdgpu_cs_chunk __user **chunk_ptr = NULL;
157 struct drm_amdgpu_cs_chunk user_chunk;
158 uint32_t __user *cdata;
159
160 chunk_ptr = u64_to_user_ptr(chunk_array[i]);
161 if (copy_from_user(&user_chunk, chunk_ptr,
162 sizeof(struct drm_amdgpu_cs_chunk))) {
163 ret = -EFAULT;
164 i--;
165 goto free_partial_kdata;
166 }
167 p->chunks[i].chunk_id = user_chunk.chunk_id;
168 p->chunks[i].length_dw = user_chunk.length_dw;
169
170 size = p->chunks[i].length_dw;
171 cdata = u64_to_user_ptr(user_chunk.chunk_data);
172
173 p->chunks[i].kdata = kvmalloc_array(size, sizeof(uint32_t), GFP_KERNEL);
174 if (p->chunks[i].kdata == NULL) {
175 ret = -ENOMEM;
176 i--;
177 goto free_partial_kdata;
178 }
179 size *= sizeof(uint32_t);
180 if (copy_from_user(p->chunks[i].kdata, cdata, size)) {
181 ret = -EFAULT;
182 goto free_partial_kdata;
183 }
184
185 switch (p->chunks[i].chunk_id) {
186 case AMDGPU_CHUNK_ID_IB:
187 ++num_ibs;
188 break;
189
190 case AMDGPU_CHUNK_ID_FENCE:
191 size = sizeof(struct drm_amdgpu_cs_chunk_fence);
192 if (p->chunks[i].length_dw * sizeof(uint32_t) < size) {
193 ret = -EINVAL;
194 goto free_partial_kdata;
195 }
196
197 ret = amdgpu_cs_user_fence_chunk(p, p->chunks[i].kdata,
198 &uf_offset);
199 if (ret)
200 goto free_partial_kdata;
201
202 break;
203
204 case AMDGPU_CHUNK_ID_BO_HANDLES:
205 size = sizeof(struct drm_amdgpu_bo_list_in);
206 if (p->chunks[i].length_dw * sizeof(uint32_t) < size) {
207 ret = -EINVAL;
208 goto free_partial_kdata;
209 }
210
211 ret = amdgpu_cs_bo_handles_chunk(p, p->chunks[i].kdata);
212 if (ret)
213 goto free_partial_kdata;
214
215 break;
216
217 case AMDGPU_CHUNK_ID_DEPENDENCIES:
218 case AMDGPU_CHUNK_ID_SYNCOBJ_IN:
219 case AMDGPU_CHUNK_ID_SYNCOBJ_OUT:
220 case AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES:
221 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_WAIT:
222 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_SIGNAL:
223 break;
224
225 default:
226 ret = -EINVAL;
227 goto free_partial_kdata;
228 }
229 }
230
231 ret = amdgpu_job_alloc(p->adev, num_ibs, &p->job, vm);
232 if (ret)
233 goto free_all_kdata;
234
235 if (p->ctx->vram_lost_counter != p->job->vram_lost_counter) {
236 ret = -ECANCELED;
237 goto free_all_kdata;
238 }
239
240 if (p->uf_entry.tv.bo)
241 p->job->uf_addr = uf_offset;
242 kfree(chunk_array);
243
244 /* Use this opportunity to fill in task info for the vm */
245 amdgpu_vm_set_task_info(vm);
246
247 return 0;
248
249free_all_kdata:
250 i = p->nchunks - 1;
251free_partial_kdata:
252 for (; i >= 0; i--)
253 kvfree(p->chunks[i].kdata);
254 kfree(p->chunks);
255 p->chunks = NULL;
256 p->nchunks = 0;
257free_chunk:
258 kfree(chunk_array);
259
260 return ret;
261}
262
263/* Convert microseconds to bytes. */
264static u64 us_to_bytes(struct amdgpu_device *adev, s64 us)
265{
266 if (us <= 0 || !adev->mm_stats.log2_max_MBps)
267 return 0;
268
269 /* Since accum_us is incremented by a million per second, just
270 * multiply it by the number of MB/s to get the number of bytes.
271 */
272 return us << adev->mm_stats.log2_max_MBps;
273}
274
275static s64 bytes_to_us(struct amdgpu_device *adev, u64 bytes)
276{
277 if (!adev->mm_stats.log2_max_MBps)
278 return 0;
279
280 return bytes >> adev->mm_stats.log2_max_MBps;
281}
282
283/* Returns how many bytes TTM can move right now. If no bytes can be moved,
284 * it returns 0. If it returns non-zero, it's OK to move at least one buffer,
285 * which means it can go over the threshold once. If that happens, the driver
286 * will be in debt and no other buffer migrations can be done until that debt
287 * is repaid.
288 *
289 * This approach allows moving a buffer of any size (it's important to allow
290 * that).
291 *
292 * The currency is simply time in microseconds and it increases as the clock
293 * ticks. The accumulated microseconds (us) are converted to bytes and
294 * returned.
295 */
296static void amdgpu_cs_get_threshold_for_moves(struct amdgpu_device *adev,
297 u64 *max_bytes,
298 u64 *max_vis_bytes)
299{
300 s64 time_us, increment_us;
301 u64 free_vram, total_vram, used_vram;
302
303 /* Allow a maximum of 200 accumulated ms. This is basically per-IB
304 * throttling.
305 *
306 * It means that in order to get full max MBps, at least 5 IBs per
307 * second must be submitted and not more than 200ms apart from each
308 * other.
309 */
310 const s64 us_upper_bound = 200000;
311
312 if (!adev->mm_stats.log2_max_MBps) {
313 *max_bytes = 0;
314 *max_vis_bytes = 0;
315 return;
316 }
317
318 total_vram = adev->gmc.real_vram_size - atomic64_read(&adev->vram_pin_size);
319 used_vram = amdgpu_vram_mgr_usage(&adev->mman.bdev.man[TTM_PL_VRAM]);
320 free_vram = used_vram >= total_vram ? 0 : total_vram - used_vram;
321
322 spin_lock(&adev->mm_stats.lock);
323
324 /* Increase the amount of accumulated us. */
325 time_us = ktime_to_us(ktime_get());
326 increment_us = time_us - adev->mm_stats.last_update_us;
327 adev->mm_stats.last_update_us = time_us;
328 adev->mm_stats.accum_us = min(adev->mm_stats.accum_us + increment_us,
329 us_upper_bound);
330
331 /* This prevents the short period of low performance when the VRAM
332 * usage is low and the driver is in debt or doesn't have enough
333 * accumulated us to fill VRAM quickly.
334 *
335 * The situation can occur in these cases:
336 * - a lot of VRAM is freed by userspace
337 * - the presence of a big buffer causes a lot of evictions
338 * (solution: split buffers into smaller ones)
339 *
340 * If 128 MB or 1/8th of VRAM is free, start filling it now by setting
341 * accum_us to a positive number.
342 */
343 if (free_vram >= 128 * 1024 * 1024 || free_vram >= total_vram / 8) {
344 s64 min_us;
345
346 /* Be more aggresive on dGPUs. Try to fill a portion of free
347 * VRAM now.
348 */
349 if (!(adev->flags & AMD_IS_APU))
350 min_us = bytes_to_us(adev, free_vram / 4);
351 else
352 min_us = 0; /* Reset accum_us on APUs. */
353
354 adev->mm_stats.accum_us = max(min_us, adev->mm_stats.accum_us);
355 }
356
357 /* This is set to 0 if the driver is in debt to disallow (optional)
358 * buffer moves.
359 */
360 *max_bytes = us_to_bytes(adev, adev->mm_stats.accum_us);
361
362 /* Do the same for visible VRAM if half of it is free */
363 if (!amdgpu_gmc_vram_full_visible(&adev->gmc)) {
364 u64 total_vis_vram = adev->gmc.visible_vram_size;
365 u64 used_vis_vram =
366 amdgpu_vram_mgr_vis_usage(&adev->mman.bdev.man[TTM_PL_VRAM]);
367
368 if (used_vis_vram < total_vis_vram) {
369 u64 free_vis_vram = total_vis_vram - used_vis_vram;
370 adev->mm_stats.accum_us_vis = min(adev->mm_stats.accum_us_vis +
371 increment_us, us_upper_bound);
372
373 if (free_vis_vram >= total_vis_vram / 2)
374 adev->mm_stats.accum_us_vis =
375 max(bytes_to_us(adev, free_vis_vram / 2),
376 adev->mm_stats.accum_us_vis);
377 }
378
379 *max_vis_bytes = us_to_bytes(adev, adev->mm_stats.accum_us_vis);
380 } else {
381 *max_vis_bytes = 0;
382 }
383
384 spin_unlock(&adev->mm_stats.lock);
385}
386
387/* Report how many bytes have really been moved for the last command
388 * submission. This can result in a debt that can stop buffer migrations
389 * temporarily.
390 */
391void amdgpu_cs_report_moved_bytes(struct amdgpu_device *adev, u64 num_bytes,
392 u64 num_vis_bytes)
393{
394 spin_lock(&adev->mm_stats.lock);
395 adev->mm_stats.accum_us -= bytes_to_us(adev, num_bytes);
396 adev->mm_stats.accum_us_vis -= bytes_to_us(adev, num_vis_bytes);
397 spin_unlock(&adev->mm_stats.lock);
398}
399
400static int amdgpu_cs_bo_validate(struct amdgpu_cs_parser *p,
401 struct amdgpu_bo *bo)
402{
403 struct amdgpu_device *adev = amdgpu_ttm_adev(bo->tbo.bdev);
404 struct ttm_operation_ctx ctx = {
405 .interruptible = true,
406 .no_wait_gpu = false,
407 .resv = bo->tbo.base.resv,
408 .flags = 0
409 };
410 uint32_t domain;
411 int r;
412
413 if (bo->pin_count)
414 return 0;
415
416 /* Don't move this buffer if we have depleted our allowance
417 * to move it. Don't move anything if the threshold is zero.
418 */
419 if (p->bytes_moved < p->bytes_moved_threshold &&
420 (!bo->tbo.base.dma_buf ||
421 list_empty(&bo->tbo.base.dma_buf->attachments))) {
422 if (!amdgpu_gmc_vram_full_visible(&adev->gmc) &&
423 (bo->flags & AMDGPU_GEM_CREATE_CPU_ACCESS_REQUIRED)) {
424 /* And don't move a CPU_ACCESS_REQUIRED BO to limited
425 * visible VRAM if we've depleted our allowance to do
426 * that.
427 */
428 if (p->bytes_moved_vis < p->bytes_moved_vis_threshold)
429 domain = bo->preferred_domains;
430 else
431 domain = bo->allowed_domains;
432 } else {
433 domain = bo->preferred_domains;
434 }
435 } else {
436 domain = bo->allowed_domains;
437 }
438
439retry:
440 amdgpu_bo_placement_from_domain(bo, domain);
441 r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
442
443 p->bytes_moved += ctx.bytes_moved;
444 if (!amdgpu_gmc_vram_full_visible(&adev->gmc) &&
445 amdgpu_bo_in_cpu_visible_vram(bo))
446 p->bytes_moved_vis += ctx.bytes_moved;
447
448 if (unlikely(r == -ENOMEM) && domain != bo->allowed_domains) {
449 domain = bo->allowed_domains;
450 goto retry;
451 }
452
453 return r;
454}
455
456static int amdgpu_cs_validate(void *param, struct amdgpu_bo *bo)
457{
458 struct amdgpu_cs_parser *p = param;
459 int r;
460
461 r = amdgpu_cs_bo_validate(p, bo);
462 if (r)
463 return r;
464
465 if (bo->shadow)
466 r = amdgpu_cs_bo_validate(p, bo->shadow);
467
468 return r;
469}
470
471static int amdgpu_cs_list_validate(struct amdgpu_cs_parser *p,
472 struct list_head *validated)
473{
474 struct ttm_operation_ctx ctx = { true, false };
475 struct amdgpu_bo_list_entry *lobj;
476 int r;
477
478 list_for_each_entry(lobj, validated, tv.head) {
479 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(lobj->tv.bo);
480 struct mm_struct *usermm;
481
482 usermm = amdgpu_ttm_tt_get_usermm(bo->tbo.ttm);
483 if (usermm && usermm != current->mm)
484 return -EPERM;
485
486 if (amdgpu_ttm_tt_is_userptr(bo->tbo.ttm) &&
487 lobj->user_invalidated && lobj->user_pages) {
488 amdgpu_bo_placement_from_domain(bo,
489 AMDGPU_GEM_DOMAIN_CPU);
490 r = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
491 if (r)
492 return r;
493
494 amdgpu_ttm_tt_set_user_pages(bo->tbo.ttm,
495 lobj->user_pages);
496 }
497
498 r = amdgpu_cs_validate(p, bo);
499 if (r)
500 return r;
501
502 kvfree(lobj->user_pages);
503 lobj->user_pages = NULL;
504 }
505 return 0;
506}
507
508static int amdgpu_cs_parser_bos(struct amdgpu_cs_parser *p,
509 union drm_amdgpu_cs *cs)
510{
511 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
512 struct amdgpu_vm *vm = &fpriv->vm;
513 struct amdgpu_bo_list_entry *e;
514 struct list_head duplicates;
515 struct amdgpu_bo *gds;
516 struct amdgpu_bo *gws;
517 struct amdgpu_bo *oa;
518 int r;
519
520 INIT_LIST_HEAD(&p->validated);
521
522 /* p->bo_list could already be assigned if AMDGPU_CHUNK_ID_BO_HANDLES is present */
523 if (cs->in.bo_list_handle) {
524 if (p->bo_list)
525 return -EINVAL;
526
527 r = amdgpu_bo_list_get(fpriv, cs->in.bo_list_handle,
528 &p->bo_list);
529 if (r)
530 return r;
531 } else if (!p->bo_list) {
532 /* Create a empty bo_list when no handle is provided */
533 r = amdgpu_bo_list_create(p->adev, p->filp, NULL, 0,
534 &p->bo_list);
535 if (r)
536 return r;
537 }
538
539 /* One for TTM and one for the CS job */
540 amdgpu_bo_list_for_each_entry(e, p->bo_list)
541 e->tv.num_shared = 2;
542
543 amdgpu_bo_list_get_list(p->bo_list, &p->validated);
544
545 INIT_LIST_HEAD(&duplicates);
546 amdgpu_vm_get_pd_bo(&fpriv->vm, &p->validated, &p->vm_pd);
547
548 if (p->uf_entry.tv.bo && !ttm_to_amdgpu_bo(p->uf_entry.tv.bo)->parent)
549 list_add(&p->uf_entry.tv.head, &p->validated);
550
551 /* Get userptr backing pages. If pages are updated after registered
552 * in amdgpu_gem_userptr_ioctl(), amdgpu_cs_list_validate() will do
553 * amdgpu_ttm_backend_bind() to flush and invalidate new pages
554 */
555 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
556 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(e->tv.bo);
557 bool userpage_invalidated = false;
558 int i;
559
560 e->user_pages = kvmalloc_array(bo->tbo.ttm->num_pages,
561 sizeof(struct page *),
562 GFP_KERNEL | __GFP_ZERO);
563 if (!e->user_pages) {
564 DRM_ERROR("calloc failure\n");
565 return -ENOMEM;
566 }
567
568 r = amdgpu_ttm_tt_get_user_pages(bo, e->user_pages);
569 if (r) {
570 kvfree(e->user_pages);
571 e->user_pages = NULL;
572 return r;
573 }
574
575 for (i = 0; i < bo->tbo.ttm->num_pages; i++) {
576 if (bo->tbo.ttm->pages[i] != e->user_pages[i]) {
577 userpage_invalidated = true;
578 break;
579 }
580 }
581 e->user_invalidated = userpage_invalidated;
582 }
583
584 r = ttm_eu_reserve_buffers(&p->ticket, &p->validated, true,
585 &duplicates);
586 if (unlikely(r != 0)) {
587 if (r != -ERESTARTSYS)
588 DRM_ERROR("ttm_eu_reserve_buffers failed.\n");
589 goto out;
590 }
591
592 amdgpu_cs_get_threshold_for_moves(p->adev, &p->bytes_moved_threshold,
593 &p->bytes_moved_vis_threshold);
594 p->bytes_moved = 0;
595 p->bytes_moved_vis = 0;
596
597 r = amdgpu_vm_validate_pt_bos(p->adev, &fpriv->vm,
598 amdgpu_cs_validate, p);
599 if (r) {
600 DRM_ERROR("amdgpu_vm_validate_pt_bos() failed.\n");
601 goto error_validate;
602 }
603
604 r = amdgpu_cs_list_validate(p, &duplicates);
605 if (r)
606 goto error_validate;
607
608 r = amdgpu_cs_list_validate(p, &p->validated);
609 if (r)
610 goto error_validate;
611
612 amdgpu_cs_report_moved_bytes(p->adev, p->bytes_moved,
613 p->bytes_moved_vis);
614
615 gds = p->bo_list->gds_obj;
616 gws = p->bo_list->gws_obj;
617 oa = p->bo_list->oa_obj;
618
619 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
620 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(e->tv.bo);
621
622 /* Make sure we use the exclusive slot for shared BOs */
623 if (bo->prime_shared_count)
624 e->tv.num_shared = 0;
625 e->bo_va = amdgpu_vm_bo_find(vm, bo);
626 }
627
628 if (gds) {
629 p->job->gds_base = amdgpu_bo_gpu_offset(gds) >> PAGE_SHIFT;
630 p->job->gds_size = amdgpu_bo_size(gds) >> PAGE_SHIFT;
631 }
632 if (gws) {
633 p->job->gws_base = amdgpu_bo_gpu_offset(gws) >> PAGE_SHIFT;
634 p->job->gws_size = amdgpu_bo_size(gws) >> PAGE_SHIFT;
635 }
636 if (oa) {
637 p->job->oa_base = amdgpu_bo_gpu_offset(oa) >> PAGE_SHIFT;
638 p->job->oa_size = amdgpu_bo_size(oa) >> PAGE_SHIFT;
639 }
640
641 if (!r && p->uf_entry.tv.bo) {
642 struct amdgpu_bo *uf = ttm_to_amdgpu_bo(p->uf_entry.tv.bo);
643
644 r = amdgpu_ttm_alloc_gart(&uf->tbo);
645 p->job->uf_addr += amdgpu_bo_gpu_offset(uf);
646 }
647
648error_validate:
649 if (r)
650 ttm_eu_backoff_reservation(&p->ticket, &p->validated);
651out:
652 return r;
653}
654
655static int amdgpu_cs_sync_rings(struct amdgpu_cs_parser *p)
656{
657 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
658 struct amdgpu_bo_list_entry *e;
659 int r;
660
661 list_for_each_entry(e, &p->validated, tv.head) {
662 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(e->tv.bo);
663 struct dma_resv *resv = bo->tbo.base.resv;
664 enum amdgpu_sync_mode sync_mode;
665
666 sync_mode = amdgpu_bo_explicit_sync(bo) ?
667 AMDGPU_SYNC_EXPLICIT : AMDGPU_SYNC_NE_OWNER;
668 r = amdgpu_sync_resv(p->adev, &p->job->sync, resv, sync_mode,
669 &fpriv->vm);
670 if (r)
671 return r;
672 }
673 return 0;
674}
675
676/**
677 * cs_parser_fini() - clean parser states
678 * @parser: parser structure holding parsing context.
679 * @error: error number
680 *
681 * If error is set than unvalidate buffer, otherwise just free memory
682 * used by parsing context.
683 **/
684static void amdgpu_cs_parser_fini(struct amdgpu_cs_parser *parser, int error,
685 bool backoff)
686{
687 unsigned i;
688
689 if (error && backoff)
690 ttm_eu_backoff_reservation(&parser->ticket,
691 &parser->validated);
692
693 for (i = 0; i < parser->num_post_deps; i++) {
694 drm_syncobj_put(parser->post_deps[i].syncobj);
695 kfree(parser->post_deps[i].chain);
696 }
697 kfree(parser->post_deps);
698
699 dma_fence_put(parser->fence);
700
701 if (parser->ctx) {
702 mutex_unlock(&parser->ctx->lock);
703 amdgpu_ctx_put(parser->ctx);
704 }
705 if (parser->bo_list)
706 amdgpu_bo_list_put(parser->bo_list);
707
708 for (i = 0; i < parser->nchunks; i++)
709 kvfree(parser->chunks[i].kdata);
710 kfree(parser->chunks);
711 if (parser->job)
712 amdgpu_job_free(parser->job);
713 if (parser->uf_entry.tv.bo) {
714 struct amdgpu_bo *uf = ttm_to_amdgpu_bo(parser->uf_entry.tv.bo);
715
716 amdgpu_bo_unref(&uf);
717 }
718}
719
720static int amdgpu_cs_vm_handling(struct amdgpu_cs_parser *p)
721{
722 struct amdgpu_ring *ring = to_amdgpu_ring(p->entity->rq->sched);
723 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
724 struct amdgpu_device *adev = p->adev;
725 struct amdgpu_vm *vm = &fpriv->vm;
726 struct amdgpu_bo_list_entry *e;
727 struct amdgpu_bo_va *bo_va;
728 struct amdgpu_bo *bo;
729 int r;
730
731 /* Only for UVD/VCE VM emulation */
732 if (ring->funcs->parse_cs || ring->funcs->patch_cs_in_place) {
733 unsigned i, j;
734
735 for (i = 0, j = 0; i < p->nchunks && j < p->job->num_ibs; i++) {
736 struct drm_amdgpu_cs_chunk_ib *chunk_ib;
737 struct amdgpu_bo_va_mapping *m;
738 struct amdgpu_bo *aobj = NULL;
739 struct amdgpu_cs_chunk *chunk;
740 uint64_t offset, va_start;
741 struct amdgpu_ib *ib;
742 uint8_t *kptr;
743
744 chunk = &p->chunks[i];
745 ib = &p->job->ibs[j];
746 chunk_ib = chunk->kdata;
747
748 if (chunk->chunk_id != AMDGPU_CHUNK_ID_IB)
749 continue;
750
751 va_start = chunk_ib->va_start & AMDGPU_GMC_HOLE_MASK;
752 r = amdgpu_cs_find_mapping(p, va_start, &aobj, &m);
753 if (r) {
754 DRM_ERROR("IB va_start is invalid\n");
755 return r;
756 }
757
758 if ((va_start + chunk_ib->ib_bytes) >
759 (m->last + 1) * AMDGPU_GPU_PAGE_SIZE) {
760 DRM_ERROR("IB va_start+ib_bytes is invalid\n");
761 return -EINVAL;
762 }
763
764 /* the IB should be reserved at this point */
765 r = amdgpu_bo_kmap(aobj, (void **)&kptr);
766 if (r) {
767 return r;
768 }
769
770 offset = m->start * AMDGPU_GPU_PAGE_SIZE;
771 kptr += va_start - offset;
772
773 if (ring->funcs->parse_cs) {
774 memcpy(ib->ptr, kptr, chunk_ib->ib_bytes);
775 amdgpu_bo_kunmap(aobj);
776
777 r = amdgpu_ring_parse_cs(ring, p, j);
778 if (r)
779 return r;
780 } else {
781 ib->ptr = (uint32_t *)kptr;
782 r = amdgpu_ring_patch_cs_in_place(ring, p, j);
783 amdgpu_bo_kunmap(aobj);
784 if (r)
785 return r;
786 }
787
788 j++;
789 }
790 }
791
792 if (!p->job->vm)
793 return amdgpu_cs_sync_rings(p);
794
795
796 r = amdgpu_vm_clear_freed(adev, vm, NULL);
797 if (r)
798 return r;
799
800 r = amdgpu_vm_bo_update(adev, fpriv->prt_va, false);
801 if (r)
802 return r;
803
804 r = amdgpu_sync_vm_fence(&p->job->sync, fpriv->prt_va->last_pt_update);
805 if (r)
806 return r;
807
808 if (amdgpu_mcbp || amdgpu_sriov_vf(adev)) {
809 bo_va = fpriv->csa_va;
810 BUG_ON(!bo_va);
811 r = amdgpu_vm_bo_update(adev, bo_va, false);
812 if (r)
813 return r;
814
815 r = amdgpu_sync_vm_fence(&p->job->sync, bo_va->last_pt_update);
816 if (r)
817 return r;
818 }
819
820 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
821 /* ignore duplicates */
822 bo = ttm_to_amdgpu_bo(e->tv.bo);
823 if (!bo)
824 continue;
825
826 bo_va = e->bo_va;
827 if (bo_va == NULL)
828 continue;
829
830 r = amdgpu_vm_bo_update(adev, bo_va, false);
831 if (r)
832 return r;
833
834 r = amdgpu_sync_vm_fence(&p->job->sync, bo_va->last_pt_update);
835 if (r)
836 return r;
837 }
838
839 r = amdgpu_vm_handle_moved(adev, vm);
840 if (r)
841 return r;
842
843 r = amdgpu_vm_update_pdes(adev, vm, false);
844 if (r)
845 return r;
846
847 r = amdgpu_sync_vm_fence(&p->job->sync, vm->last_update);
848 if (r)
849 return r;
850
851 p->job->vm_pd_addr = amdgpu_gmc_pd_addr(vm->root.base.bo);
852
853 if (amdgpu_vm_debug) {
854 /* Invalidate all BOs to test for userspace bugs */
855 amdgpu_bo_list_for_each_entry(e, p->bo_list) {
856 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(e->tv.bo);
857
858 /* ignore duplicates */
859 if (!bo)
860 continue;
861
862 amdgpu_vm_bo_invalidate(adev, bo, false);
863 }
864 }
865
866 return amdgpu_cs_sync_rings(p);
867}
868
869static int amdgpu_cs_ib_fill(struct amdgpu_device *adev,
870 struct amdgpu_cs_parser *parser)
871{
872 struct amdgpu_fpriv *fpriv = parser->filp->driver_priv;
873 struct amdgpu_vm *vm = &fpriv->vm;
874 int r, ce_preempt = 0, de_preempt = 0;
875 struct amdgpu_ring *ring;
876 int i, j;
877
878 for (i = 0, j = 0; i < parser->nchunks && j < parser->job->num_ibs; i++) {
879 struct amdgpu_cs_chunk *chunk;
880 struct amdgpu_ib *ib;
881 struct drm_amdgpu_cs_chunk_ib *chunk_ib;
882 struct drm_sched_entity *entity;
883
884 chunk = &parser->chunks[i];
885 ib = &parser->job->ibs[j];
886 chunk_ib = (struct drm_amdgpu_cs_chunk_ib *)chunk->kdata;
887
888 if (chunk->chunk_id != AMDGPU_CHUNK_ID_IB)
889 continue;
890
891 if (chunk_ib->ip_type == AMDGPU_HW_IP_GFX &&
892 (amdgpu_mcbp || amdgpu_sriov_vf(adev))) {
893 if (chunk_ib->flags & AMDGPU_IB_FLAG_PREEMPT) {
894 if (chunk_ib->flags & AMDGPU_IB_FLAG_CE)
895 ce_preempt++;
896 else
897 de_preempt++;
898 }
899
900 /* each GFX command submit allows 0 or 1 IB preemptible for CE & DE */
901 if (ce_preempt > 1 || de_preempt > 1)
902 return -EINVAL;
903 }
904
905 r = amdgpu_ctx_get_entity(parser->ctx, chunk_ib->ip_type,
906 chunk_ib->ip_instance, chunk_ib->ring,
907 &entity);
908 if (r)
909 return r;
910
911 if (chunk_ib->flags & AMDGPU_IB_FLAG_PREAMBLE)
912 parser->job->preamble_status |=
913 AMDGPU_PREAMBLE_IB_PRESENT;
914
915 if (parser->entity && parser->entity != entity)
916 return -EINVAL;
917
918 /* Return if there is no run queue associated with this entity.
919 * Possibly because of disabled HW IP*/
920 if (entity->rq == NULL)
921 return -EINVAL;
922
923 parser->entity = entity;
924
925 ring = to_amdgpu_ring(entity->rq->sched);
926 r = amdgpu_ib_get(adev, vm, ring->funcs->parse_cs ?
927 chunk_ib->ib_bytes : 0,
928 AMDGPU_IB_POOL_DELAYED, ib);
929 if (r) {
930 DRM_ERROR("Failed to get ib !\n");
931 return r;
932 }
933
934 ib->gpu_addr = chunk_ib->va_start;
935 ib->length_dw = chunk_ib->ib_bytes / 4;
936 ib->flags = chunk_ib->flags;
937
938 j++;
939 }
940
941 /* MM engine doesn't support user fences */
942 ring = to_amdgpu_ring(parser->entity->rq->sched);
943 if (parser->job->uf_addr && ring->funcs->no_user_fence)
944 return -EINVAL;
945
946 return amdgpu_ctx_wait_prev_fence(parser->ctx, parser->entity);
947}
948
949static int amdgpu_cs_process_fence_dep(struct amdgpu_cs_parser *p,
950 struct amdgpu_cs_chunk *chunk)
951{
952 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
953 unsigned num_deps;
954 int i, r;
955 struct drm_amdgpu_cs_chunk_dep *deps;
956
957 deps = (struct drm_amdgpu_cs_chunk_dep *)chunk->kdata;
958 num_deps = chunk->length_dw * 4 /
959 sizeof(struct drm_amdgpu_cs_chunk_dep);
960
961 for (i = 0; i < num_deps; ++i) {
962 struct amdgpu_ctx *ctx;
963 struct drm_sched_entity *entity;
964 struct dma_fence *fence;
965
966 ctx = amdgpu_ctx_get(fpriv, deps[i].ctx_id);
967 if (ctx == NULL)
968 return -EINVAL;
969
970 r = amdgpu_ctx_get_entity(ctx, deps[i].ip_type,
971 deps[i].ip_instance,
972 deps[i].ring, &entity);
973 if (r) {
974 amdgpu_ctx_put(ctx);
975 return r;
976 }
977
978 fence = amdgpu_ctx_get_fence(ctx, entity, deps[i].handle);
979 amdgpu_ctx_put(ctx);
980
981 if (IS_ERR(fence))
982 return PTR_ERR(fence);
983 else if (!fence)
984 continue;
985
986 if (chunk->chunk_id == AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES) {
987 struct drm_sched_fence *s_fence;
988 struct dma_fence *old = fence;
989
990 s_fence = to_drm_sched_fence(fence);
991 fence = dma_fence_get(&s_fence->scheduled);
992 dma_fence_put(old);
993 }
994
995 r = amdgpu_sync_fence(&p->job->sync, fence);
996 dma_fence_put(fence);
997 if (r)
998 return r;
999 }
1000 return 0;
1001}
1002
1003static int amdgpu_syncobj_lookup_and_add_to_sync(struct amdgpu_cs_parser *p,
1004 uint32_t handle, u64 point,
1005 u64 flags)
1006{
1007 struct dma_fence *fence;
1008 int r;
1009
1010 r = drm_syncobj_find_fence(p->filp, handle, point, flags, &fence);
1011 if (r) {
1012 DRM_ERROR("syncobj %u failed to find fence @ %llu (%d)!\n",
1013 handle, point, r);
1014 return r;
1015 }
1016
1017 r = amdgpu_sync_fence(&p->job->sync, fence);
1018 dma_fence_put(fence);
1019
1020 return r;
1021}
1022
1023static int amdgpu_cs_process_syncobj_in_dep(struct amdgpu_cs_parser *p,
1024 struct amdgpu_cs_chunk *chunk)
1025{
1026 struct drm_amdgpu_cs_chunk_sem *deps;
1027 unsigned num_deps;
1028 int i, r;
1029
1030 deps = (struct drm_amdgpu_cs_chunk_sem *)chunk->kdata;
1031 num_deps = chunk->length_dw * 4 /
1032 sizeof(struct drm_amdgpu_cs_chunk_sem);
1033 for (i = 0; i < num_deps; ++i) {
1034 r = amdgpu_syncobj_lookup_and_add_to_sync(p, deps[i].handle,
1035 0, 0);
1036 if (r)
1037 return r;
1038 }
1039
1040 return 0;
1041}
1042
1043
1044static int amdgpu_cs_process_syncobj_timeline_in_dep(struct amdgpu_cs_parser *p,
1045 struct amdgpu_cs_chunk *chunk)
1046{
1047 struct drm_amdgpu_cs_chunk_syncobj *syncobj_deps;
1048 unsigned num_deps;
1049 int i, r;
1050
1051 syncobj_deps = (struct drm_amdgpu_cs_chunk_syncobj *)chunk->kdata;
1052 num_deps = chunk->length_dw * 4 /
1053 sizeof(struct drm_amdgpu_cs_chunk_syncobj);
1054 for (i = 0; i < num_deps; ++i) {
1055 r = amdgpu_syncobj_lookup_and_add_to_sync(p,
1056 syncobj_deps[i].handle,
1057 syncobj_deps[i].point,
1058 syncobj_deps[i].flags);
1059 if (r)
1060 return r;
1061 }
1062
1063 return 0;
1064}
1065
1066static int amdgpu_cs_process_syncobj_out_dep(struct amdgpu_cs_parser *p,
1067 struct amdgpu_cs_chunk *chunk)
1068{
1069 struct drm_amdgpu_cs_chunk_sem *deps;
1070 unsigned num_deps;
1071 int i;
1072
1073 deps = (struct drm_amdgpu_cs_chunk_sem *)chunk->kdata;
1074 num_deps = chunk->length_dw * 4 /
1075 sizeof(struct drm_amdgpu_cs_chunk_sem);
1076
1077 if (p->post_deps)
1078 return -EINVAL;
1079
1080 p->post_deps = kmalloc_array(num_deps, sizeof(*p->post_deps),
1081 GFP_KERNEL);
1082 p->num_post_deps = 0;
1083
1084 if (!p->post_deps)
1085 return -ENOMEM;
1086
1087
1088 for (i = 0; i < num_deps; ++i) {
1089 p->post_deps[i].syncobj =
1090 drm_syncobj_find(p->filp, deps[i].handle);
1091 if (!p->post_deps[i].syncobj)
1092 return -EINVAL;
1093 p->post_deps[i].chain = NULL;
1094 p->post_deps[i].point = 0;
1095 p->num_post_deps++;
1096 }
1097
1098 return 0;
1099}
1100
1101
1102static int amdgpu_cs_process_syncobj_timeline_out_dep(struct amdgpu_cs_parser *p,
1103 struct amdgpu_cs_chunk *chunk)
1104{
1105 struct drm_amdgpu_cs_chunk_syncobj *syncobj_deps;
1106 unsigned num_deps;
1107 int i;
1108
1109 syncobj_deps = (struct drm_amdgpu_cs_chunk_syncobj *)chunk->kdata;
1110 num_deps = chunk->length_dw * 4 /
1111 sizeof(struct drm_amdgpu_cs_chunk_syncobj);
1112
1113 if (p->post_deps)
1114 return -EINVAL;
1115
1116 p->post_deps = kmalloc_array(num_deps, sizeof(*p->post_deps),
1117 GFP_KERNEL);
1118 p->num_post_deps = 0;
1119
1120 if (!p->post_deps)
1121 return -ENOMEM;
1122
1123 for (i = 0; i < num_deps; ++i) {
1124 struct amdgpu_cs_post_dep *dep = &p->post_deps[i];
1125
1126 dep->chain = NULL;
1127 if (syncobj_deps[i].point) {
1128 dep->chain = kmalloc(sizeof(*dep->chain), GFP_KERNEL);
1129 if (!dep->chain)
1130 return -ENOMEM;
1131 }
1132
1133 dep->syncobj = drm_syncobj_find(p->filp,
1134 syncobj_deps[i].handle);
1135 if (!dep->syncobj) {
1136 kfree(dep->chain);
1137 return -EINVAL;
1138 }
1139 dep->point = syncobj_deps[i].point;
1140 p->num_post_deps++;
1141 }
1142
1143 return 0;
1144}
1145
1146static int amdgpu_cs_dependencies(struct amdgpu_device *adev,
1147 struct amdgpu_cs_parser *p)
1148{
1149 int i, r;
1150
1151 for (i = 0; i < p->nchunks; ++i) {
1152 struct amdgpu_cs_chunk *chunk;
1153
1154 chunk = &p->chunks[i];
1155
1156 switch (chunk->chunk_id) {
1157 case AMDGPU_CHUNK_ID_DEPENDENCIES:
1158 case AMDGPU_CHUNK_ID_SCHEDULED_DEPENDENCIES:
1159 r = amdgpu_cs_process_fence_dep(p, chunk);
1160 if (r)
1161 return r;
1162 break;
1163 case AMDGPU_CHUNK_ID_SYNCOBJ_IN:
1164 r = amdgpu_cs_process_syncobj_in_dep(p, chunk);
1165 if (r)
1166 return r;
1167 break;
1168 case AMDGPU_CHUNK_ID_SYNCOBJ_OUT:
1169 r = amdgpu_cs_process_syncobj_out_dep(p, chunk);
1170 if (r)
1171 return r;
1172 break;
1173 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_WAIT:
1174 r = amdgpu_cs_process_syncobj_timeline_in_dep(p, chunk);
1175 if (r)
1176 return r;
1177 break;
1178 case AMDGPU_CHUNK_ID_SYNCOBJ_TIMELINE_SIGNAL:
1179 r = amdgpu_cs_process_syncobj_timeline_out_dep(p, chunk);
1180 if (r)
1181 return r;
1182 break;
1183 }
1184 }
1185
1186 return 0;
1187}
1188
1189static void amdgpu_cs_post_dependencies(struct amdgpu_cs_parser *p)
1190{
1191 int i;
1192
1193 for (i = 0; i < p->num_post_deps; ++i) {
1194 if (p->post_deps[i].chain && p->post_deps[i].point) {
1195 drm_syncobj_add_point(p->post_deps[i].syncobj,
1196 p->post_deps[i].chain,
1197 p->fence, p->post_deps[i].point);
1198 p->post_deps[i].chain = NULL;
1199 } else {
1200 drm_syncobj_replace_fence(p->post_deps[i].syncobj,
1201 p->fence);
1202 }
1203 }
1204}
1205
1206static int amdgpu_cs_submit(struct amdgpu_cs_parser *p,
1207 union drm_amdgpu_cs *cs)
1208{
1209 struct amdgpu_fpriv *fpriv = p->filp->driver_priv;
1210 struct drm_sched_entity *entity = p->entity;
1211 struct amdgpu_bo_list_entry *e;
1212 struct amdgpu_job *job;
1213 uint64_t seq;
1214 int r;
1215
1216 job = p->job;
1217 p->job = NULL;
1218
1219 r = drm_sched_job_init(&job->base, entity, &fpriv->vm);
1220 if (r)
1221 goto error_unlock;
1222
1223 /* No memory allocation is allowed while holding the notifier lock.
1224 * The lock is held until amdgpu_cs_submit is finished and fence is
1225 * added to BOs.
1226 */
1227 mutex_lock(&p->adev->notifier_lock);
1228
1229 /* If userptr are invalidated after amdgpu_cs_parser_bos(), return
1230 * -EAGAIN, drmIoctl in libdrm will restart the amdgpu_cs_ioctl.
1231 */
1232 amdgpu_bo_list_for_each_userptr_entry(e, p->bo_list) {
1233 struct amdgpu_bo *bo = ttm_to_amdgpu_bo(e->tv.bo);
1234
1235 r |= !amdgpu_ttm_tt_get_user_pages_done(bo->tbo.ttm);
1236 }
1237 if (r) {
1238 r = -EAGAIN;
1239 goto error_abort;
1240 }
1241
1242 p->fence = dma_fence_get(&job->base.s_fence->finished);
1243
1244 amdgpu_ctx_add_fence(p->ctx, entity, p->fence, &seq);
1245 amdgpu_cs_post_dependencies(p);
1246
1247 if ((job->preamble_status & AMDGPU_PREAMBLE_IB_PRESENT) &&
1248 !p->ctx->preamble_presented) {
1249 job->preamble_status |= AMDGPU_PREAMBLE_IB_PRESENT_FIRST;
1250 p->ctx->preamble_presented = true;
1251 }
1252
1253 cs->out.handle = seq;
1254 job->uf_sequence = seq;
1255
1256 amdgpu_job_free_resources(job);
1257
1258 trace_amdgpu_cs_ioctl(job);
1259 amdgpu_vm_bo_trace_cs(&fpriv->vm, &p->ticket);
1260 drm_sched_entity_push_job(&job->base, entity);
1261
1262 amdgpu_vm_move_to_lru_tail(p->adev, &fpriv->vm);
1263
1264 ttm_eu_fence_buffer_objects(&p->ticket, &p->validated, p->fence);
1265 mutex_unlock(&p->adev->notifier_lock);
1266
1267 return 0;
1268
1269error_abort:
1270 drm_sched_job_cleanup(&job->base);
1271 mutex_unlock(&p->adev->notifier_lock);
1272
1273error_unlock:
1274 amdgpu_job_free(job);
1275 return r;
1276}
1277
1278int amdgpu_cs_ioctl(struct drm_device *dev, void *data, struct drm_file *filp)
1279{
1280 struct amdgpu_device *adev = dev->dev_private;
1281 union drm_amdgpu_cs *cs = data;
1282 struct amdgpu_cs_parser parser = {};
1283 bool reserved_buffers = false;
1284 int i, r;
1285
1286 if (amdgpu_ras_intr_triggered())
1287 return -EHWPOISON;
1288
1289 if (!adev->accel_working)
1290 return -EBUSY;
1291
1292 parser.adev = adev;
1293 parser.filp = filp;
1294
1295 r = amdgpu_cs_parser_init(&parser, data);
1296 if (r) {
1297 DRM_ERROR("Failed to initialize parser %d!\n", r);
1298 goto out;
1299 }
1300
1301 r = amdgpu_cs_ib_fill(adev, &parser);
1302 if (r)
1303 goto out;
1304
1305 r = amdgpu_cs_dependencies(adev, &parser);
1306 if (r) {
1307 DRM_ERROR("Failed in the dependencies handling %d!\n", r);
1308 goto out;
1309 }
1310
1311 r = amdgpu_cs_parser_bos(&parser, data);
1312 if (r) {
1313 if (r == -ENOMEM)
1314 DRM_ERROR("Not enough memory for command submission!\n");
1315 else if (r != -ERESTARTSYS && r != -EAGAIN)
1316 DRM_ERROR("Failed to process the buffer list %d!\n", r);
1317 goto out;
1318 }
1319
1320 reserved_buffers = true;
1321
1322 for (i = 0; i < parser.job->num_ibs; i++)
1323 trace_amdgpu_cs(&parser, i);
1324
1325 r = amdgpu_cs_vm_handling(&parser);
1326 if (r)
1327 goto out;
1328
1329 r = amdgpu_cs_submit(&parser, cs);
1330
1331out:
1332 amdgpu_cs_parser_fini(&parser, r, reserved_buffers);
1333
1334 return r;
1335}
1336
1337/**
1338 * amdgpu_cs_wait_ioctl - wait for a command submission to finish
1339 *
1340 * @dev: drm device
1341 * @data: data from userspace
1342 * @filp: file private
1343 *
1344 * Wait for the command submission identified by handle to finish.
1345 */
1346int amdgpu_cs_wait_ioctl(struct drm_device *dev, void *data,
1347 struct drm_file *filp)
1348{
1349 union drm_amdgpu_wait_cs *wait = data;
1350 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout);
1351 struct drm_sched_entity *entity;
1352 struct amdgpu_ctx *ctx;
1353 struct dma_fence *fence;
1354 long r;
1355
1356 ctx = amdgpu_ctx_get(filp->driver_priv, wait->in.ctx_id);
1357 if (ctx == NULL)
1358 return -EINVAL;
1359
1360 r = amdgpu_ctx_get_entity(ctx, wait->in.ip_type, wait->in.ip_instance,
1361 wait->in.ring, &entity);
1362 if (r) {
1363 amdgpu_ctx_put(ctx);
1364 return r;
1365 }
1366
1367 fence = amdgpu_ctx_get_fence(ctx, entity, wait->in.handle);
1368 if (IS_ERR(fence))
1369 r = PTR_ERR(fence);
1370 else if (fence) {
1371 r = dma_fence_wait_timeout(fence, true, timeout);
1372 if (r > 0 && fence->error)
1373 r = fence->error;
1374 dma_fence_put(fence);
1375 } else
1376 r = 1;
1377
1378 amdgpu_ctx_put(ctx);
1379 if (r < 0)
1380 return r;
1381
1382 memset(wait, 0, sizeof(*wait));
1383 wait->out.status = (r == 0);
1384
1385 return 0;
1386}
1387
1388/**
1389 * amdgpu_cs_get_fence - helper to get fence from drm_amdgpu_fence
1390 *
1391 * @adev: amdgpu device
1392 * @filp: file private
1393 * @user: drm_amdgpu_fence copied from user space
1394 */
1395static struct dma_fence *amdgpu_cs_get_fence(struct amdgpu_device *adev,
1396 struct drm_file *filp,
1397 struct drm_amdgpu_fence *user)
1398{
1399 struct drm_sched_entity *entity;
1400 struct amdgpu_ctx *ctx;
1401 struct dma_fence *fence;
1402 int r;
1403
1404 ctx = amdgpu_ctx_get(filp->driver_priv, user->ctx_id);
1405 if (ctx == NULL)
1406 return ERR_PTR(-EINVAL);
1407
1408 r = amdgpu_ctx_get_entity(ctx, user->ip_type, user->ip_instance,
1409 user->ring, &entity);
1410 if (r) {
1411 amdgpu_ctx_put(ctx);
1412 return ERR_PTR(r);
1413 }
1414
1415 fence = amdgpu_ctx_get_fence(ctx, entity, user->seq_no);
1416 amdgpu_ctx_put(ctx);
1417
1418 return fence;
1419}
1420
1421int amdgpu_cs_fence_to_handle_ioctl(struct drm_device *dev, void *data,
1422 struct drm_file *filp)
1423{
1424 struct amdgpu_device *adev = dev->dev_private;
1425 union drm_amdgpu_fence_to_handle *info = data;
1426 struct dma_fence *fence;
1427 struct drm_syncobj *syncobj;
1428 struct sync_file *sync_file;
1429 int fd, r;
1430
1431 fence = amdgpu_cs_get_fence(adev, filp, &info->in.fence);
1432 if (IS_ERR(fence))
1433 return PTR_ERR(fence);
1434
1435 if (!fence)
1436 fence = dma_fence_get_stub();
1437
1438 switch (info->in.what) {
1439 case AMDGPU_FENCE_TO_HANDLE_GET_SYNCOBJ:
1440 r = drm_syncobj_create(&syncobj, 0, fence);
1441 dma_fence_put(fence);
1442 if (r)
1443 return r;
1444 r = drm_syncobj_get_handle(filp, syncobj, &info->out.handle);
1445 drm_syncobj_put(syncobj);
1446 return r;
1447
1448 case AMDGPU_FENCE_TO_HANDLE_GET_SYNCOBJ_FD:
1449 r = drm_syncobj_create(&syncobj, 0, fence);
1450 dma_fence_put(fence);
1451 if (r)
1452 return r;
1453 r = drm_syncobj_get_fd(syncobj, (int*)&info->out.handle);
1454 drm_syncobj_put(syncobj);
1455 return r;
1456
1457 case AMDGPU_FENCE_TO_HANDLE_GET_SYNC_FILE_FD:
1458 fd = get_unused_fd_flags(O_CLOEXEC);
1459 if (fd < 0) {
1460 dma_fence_put(fence);
1461 return fd;
1462 }
1463
1464 sync_file = sync_file_create(fence);
1465 dma_fence_put(fence);
1466 if (!sync_file) {
1467 put_unused_fd(fd);
1468 return -ENOMEM;
1469 }
1470
1471 fd_install(fd, sync_file->file);
1472 info->out.handle = fd;
1473 return 0;
1474
1475 default:
1476 return -EINVAL;
1477 }
1478}
1479
1480/**
1481 * amdgpu_cs_wait_all_fence - wait on all fences to signal
1482 *
1483 * @adev: amdgpu device
1484 * @filp: file private
1485 * @wait: wait parameters
1486 * @fences: array of drm_amdgpu_fence
1487 */
1488static int amdgpu_cs_wait_all_fences(struct amdgpu_device *adev,
1489 struct drm_file *filp,
1490 union drm_amdgpu_wait_fences *wait,
1491 struct drm_amdgpu_fence *fences)
1492{
1493 uint32_t fence_count = wait->in.fence_count;
1494 unsigned int i;
1495 long r = 1;
1496
1497 for (i = 0; i < fence_count; i++) {
1498 struct dma_fence *fence;
1499 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout_ns);
1500
1501 fence = amdgpu_cs_get_fence(adev, filp, &fences[i]);
1502 if (IS_ERR(fence))
1503 return PTR_ERR(fence);
1504 else if (!fence)
1505 continue;
1506
1507 r = dma_fence_wait_timeout(fence, true, timeout);
1508 dma_fence_put(fence);
1509 if (r < 0)
1510 return r;
1511
1512 if (r == 0)
1513 break;
1514
1515 if (fence->error)
1516 return fence->error;
1517 }
1518
1519 memset(wait, 0, sizeof(*wait));
1520 wait->out.status = (r > 0);
1521
1522 return 0;
1523}
1524
1525/**
1526 * amdgpu_cs_wait_any_fence - wait on any fence to signal
1527 *
1528 * @adev: amdgpu device
1529 * @filp: file private
1530 * @wait: wait parameters
1531 * @fences: array of drm_amdgpu_fence
1532 */
1533static int amdgpu_cs_wait_any_fence(struct amdgpu_device *adev,
1534 struct drm_file *filp,
1535 union drm_amdgpu_wait_fences *wait,
1536 struct drm_amdgpu_fence *fences)
1537{
1538 unsigned long timeout = amdgpu_gem_timeout(wait->in.timeout_ns);
1539 uint32_t fence_count = wait->in.fence_count;
1540 uint32_t first = ~0;
1541 struct dma_fence **array;
1542 unsigned int i;
1543 long r;
1544
1545 /* Prepare the fence array */
1546 array = kcalloc(fence_count, sizeof(struct dma_fence *), GFP_KERNEL);
1547
1548 if (array == NULL)
1549 return -ENOMEM;
1550
1551 for (i = 0; i < fence_count; i++) {
1552 struct dma_fence *fence;
1553
1554 fence = amdgpu_cs_get_fence(adev, filp, &fences[i]);
1555 if (IS_ERR(fence)) {
1556 r = PTR_ERR(fence);
1557 goto err_free_fence_array;
1558 } else if (fence) {
1559 array[i] = fence;
1560 } else { /* NULL, the fence has been already signaled */
1561 r = 1;
1562 first = i;
1563 goto out;
1564 }
1565 }
1566
1567 r = dma_fence_wait_any_timeout(array, fence_count, true, timeout,
1568 &first);
1569 if (r < 0)
1570 goto err_free_fence_array;
1571
1572out:
1573 memset(wait, 0, sizeof(*wait));
1574 wait->out.status = (r > 0);
1575 wait->out.first_signaled = first;
1576
1577 if (first < fence_count && array[first])
1578 r = array[first]->error;
1579 else
1580 r = 0;
1581
1582err_free_fence_array:
1583 for (i = 0; i < fence_count; i++)
1584 dma_fence_put(array[i]);
1585 kfree(array);
1586
1587 return r;
1588}
1589
1590/**
1591 * amdgpu_cs_wait_fences_ioctl - wait for multiple command submissions to finish
1592 *
1593 * @dev: drm device
1594 * @data: data from userspace
1595 * @filp: file private
1596 */
1597int amdgpu_cs_wait_fences_ioctl(struct drm_device *dev, void *data,
1598 struct drm_file *filp)
1599{
1600 struct amdgpu_device *adev = dev->dev_private;
1601 union drm_amdgpu_wait_fences *wait = data;
1602 uint32_t fence_count = wait->in.fence_count;
1603 struct drm_amdgpu_fence *fences_user;
1604 struct drm_amdgpu_fence *fences;
1605 int r;
1606
1607 /* Get the fences from userspace */
1608 fences = kmalloc_array(fence_count, sizeof(struct drm_amdgpu_fence),
1609 GFP_KERNEL);
1610 if (fences == NULL)
1611 return -ENOMEM;
1612
1613 fences_user = u64_to_user_ptr(wait->in.fences);
1614 if (copy_from_user(fences, fences_user,
1615 sizeof(struct drm_amdgpu_fence) * fence_count)) {
1616 r = -EFAULT;
1617 goto err_free_fences;
1618 }
1619
1620 if (wait->in.wait_all)
1621 r = amdgpu_cs_wait_all_fences(adev, filp, wait, fences);
1622 else
1623 r = amdgpu_cs_wait_any_fence(adev, filp, wait, fences);
1624
1625err_free_fences:
1626 kfree(fences);
1627
1628 return r;
1629}
1630
1631/**
1632 * amdgpu_cs_find_bo_va - find bo_va for VM address
1633 *
1634 * @parser: command submission parser context
1635 * @addr: VM address
1636 * @bo: resulting BO of the mapping found
1637 *
1638 * Search the buffer objects in the command submission context for a certain
1639 * virtual memory address. Returns allocation structure when found, NULL
1640 * otherwise.
1641 */
1642int amdgpu_cs_find_mapping(struct amdgpu_cs_parser *parser,
1643 uint64_t addr, struct amdgpu_bo **bo,
1644 struct amdgpu_bo_va_mapping **map)
1645{
1646 struct amdgpu_fpriv *fpriv = parser->filp->driver_priv;
1647 struct ttm_operation_ctx ctx = { false, false };
1648 struct amdgpu_vm *vm = &fpriv->vm;
1649 struct amdgpu_bo_va_mapping *mapping;
1650 int r;
1651
1652 addr /= AMDGPU_GPU_PAGE_SIZE;
1653
1654 mapping = amdgpu_vm_bo_lookup_mapping(vm, addr);
1655 if (!mapping || !mapping->bo_va || !mapping->bo_va->base.bo)
1656 return -EINVAL;
1657
1658 *bo = mapping->bo_va->base.bo;
1659 *map = mapping;
1660
1661 /* Double check that the BO is reserved by this CS */
1662 if (dma_resv_locking_ctx((*bo)->tbo.base.resv) != &parser->ticket)
1663 return -EINVAL;
1664
1665 if (!((*bo)->flags & AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS)) {
1666 (*bo)->flags |= AMDGPU_GEM_CREATE_VRAM_CONTIGUOUS;
1667 amdgpu_bo_placement_from_domain(*bo, (*bo)->allowed_domains);
1668 r = ttm_bo_validate(&(*bo)->tbo, &(*bo)->placement, &ctx);
1669 if (r)
1670 return r;
1671 }
1672
1673 return amdgpu_ttm_alloc_gart(&(*bo)->tbo);
1674}