OSDN Git Service

tracing/synthetic: Make lastcmd_mutex static
[tomoyo/tomoyo-test1.git] / drivers / accel / ivpu / ivpu_job.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (C) 2020-2023 Intel Corporation
4  */
5
6 #include <drm/drm_file.h>
7
8 #include <linux/bitfield.h>
9 #include <linux/highmem.h>
10 #include <linux/kthread.h>
11 #include <linux/pci.h>
12 #include <linux/module.h>
13 #include <uapi/drm/ivpu_accel.h>
14
15 #include "ivpu_drv.h"
16 #include "ivpu_hw.h"
17 #include "ivpu_ipc.h"
18 #include "ivpu_job.h"
19 #include "ivpu_jsm_msg.h"
20 #include "ivpu_pm.h"
21
22 #define CMD_BUF_IDX          0
23 #define JOB_ID_JOB_MASK      GENMASK(7, 0)
24 #define JOB_ID_CONTEXT_MASK  GENMASK(31, 8)
25 #define JOB_MAX_BUFFER_COUNT 65535
26
27 static unsigned int ivpu_tdr_timeout_ms;
28 module_param_named(tdr_timeout_ms, ivpu_tdr_timeout_ms, uint, 0644);
29 MODULE_PARM_DESC(tdr_timeout_ms, "Timeout for device hang detection, in milliseconds, 0 - default");
30
31 static void ivpu_cmdq_ring_db(struct ivpu_device *vdev, struct ivpu_cmdq *cmdq)
32 {
33         ivpu_hw_reg_db_set(vdev, cmdq->db_id);
34 }
35
36 static struct ivpu_cmdq *ivpu_cmdq_alloc(struct ivpu_file_priv *file_priv, u16 engine)
37 {
38         struct ivpu_device *vdev = file_priv->vdev;
39         struct vpu_job_queue_header *jobq_header;
40         struct ivpu_cmdq *cmdq;
41
42         cmdq = kzalloc(sizeof(*cmdq), GFP_KERNEL);
43         if (!cmdq)
44                 return NULL;
45
46         cmdq->mem = ivpu_bo_alloc_internal(vdev, 0, SZ_4K, DRM_IVPU_BO_WC);
47         if (!cmdq->mem)
48                 goto cmdq_free;
49
50         cmdq->db_id = file_priv->ctx.id + engine * ivpu_get_context_count(vdev);
51         cmdq->entry_count = (u32)((cmdq->mem->base.size - sizeof(struct vpu_job_queue_header)) /
52                                   sizeof(struct vpu_job_queue_entry));
53
54         cmdq->jobq = (struct vpu_job_queue *)cmdq->mem->kvaddr;
55         jobq_header = &cmdq->jobq->header;
56         jobq_header->engine_idx = engine;
57         jobq_header->head = 0;
58         jobq_header->tail = 0;
59         wmb(); /* Flush WC buffer for jobq->header */
60
61         return cmdq;
62
63 cmdq_free:
64         kfree(cmdq);
65         return NULL;
66 }
67
68 static void ivpu_cmdq_free(struct ivpu_file_priv *file_priv, struct ivpu_cmdq *cmdq)
69 {
70         if (!cmdq)
71                 return;
72
73         ivpu_bo_free_internal(cmdq->mem);
74         kfree(cmdq);
75 }
76
77 static struct ivpu_cmdq *ivpu_cmdq_acquire(struct ivpu_file_priv *file_priv, u16 engine)
78 {
79         struct ivpu_device *vdev = file_priv->vdev;
80         struct ivpu_cmdq *cmdq = file_priv->cmdq[engine];
81         int ret;
82
83         lockdep_assert_held(&file_priv->lock);
84
85         if (!cmdq) {
86                 cmdq = ivpu_cmdq_alloc(file_priv, engine);
87                 if (!cmdq)
88                         return NULL;
89                 file_priv->cmdq[engine] = cmdq;
90         }
91
92         if (cmdq->db_registered)
93                 return cmdq;
94
95         ret = ivpu_jsm_register_db(vdev, file_priv->ctx.id, cmdq->db_id,
96                                    cmdq->mem->vpu_addr, cmdq->mem->base.size);
97         if (ret)
98                 return NULL;
99
100         cmdq->db_registered = true;
101
102         return cmdq;
103 }
104
105 static void ivpu_cmdq_release_locked(struct ivpu_file_priv *file_priv, u16 engine)
106 {
107         struct ivpu_cmdq *cmdq = file_priv->cmdq[engine];
108
109         lockdep_assert_held(&file_priv->lock);
110
111         if (cmdq) {
112                 file_priv->cmdq[engine] = NULL;
113                 if (cmdq->db_registered)
114                         ivpu_jsm_unregister_db(file_priv->vdev, cmdq->db_id);
115
116                 ivpu_cmdq_free(file_priv, cmdq);
117         }
118 }
119
120 void ivpu_cmdq_release_all(struct ivpu_file_priv *file_priv)
121 {
122         int i;
123
124         mutex_lock(&file_priv->lock);
125
126         for (i = 0; i < IVPU_NUM_ENGINES; i++)
127                 ivpu_cmdq_release_locked(file_priv, i);
128
129         mutex_unlock(&file_priv->lock);
130 }
131
132 /*
133  * Mark the doorbell as unregistered and reset job queue pointers.
134  * This function needs to be called when the VPU hardware is restarted
135  * and FW looses job queue state. The next time job queue is used it
136  * will be registered again.
137  */
138 static void ivpu_cmdq_reset_locked(struct ivpu_file_priv *file_priv, u16 engine)
139 {
140         struct ivpu_cmdq *cmdq = file_priv->cmdq[engine];
141
142         lockdep_assert_held(&file_priv->lock);
143
144         if (cmdq) {
145                 cmdq->db_registered = false;
146                 cmdq->jobq->header.head = 0;
147                 cmdq->jobq->header.tail = 0;
148                 wmb(); /* Flush WC buffer for jobq header */
149         }
150 }
151
152 static void ivpu_cmdq_reset_all(struct ivpu_file_priv *file_priv)
153 {
154         int i;
155
156         mutex_lock(&file_priv->lock);
157
158         for (i = 0; i < IVPU_NUM_ENGINES; i++)
159                 ivpu_cmdq_reset_locked(file_priv, i);
160
161         mutex_unlock(&file_priv->lock);
162 }
163
164 void ivpu_cmdq_reset_all_contexts(struct ivpu_device *vdev)
165 {
166         struct ivpu_file_priv *file_priv;
167         unsigned long ctx_id;
168
169         xa_for_each(&vdev->context_xa, ctx_id, file_priv) {
170                 file_priv = ivpu_file_priv_get_by_ctx_id(vdev, ctx_id);
171                 if (!file_priv)
172                         continue;
173
174                 ivpu_cmdq_reset_all(file_priv);
175
176                 ivpu_file_priv_put(&file_priv);
177         }
178 }
179
180 static int ivpu_cmdq_push_job(struct ivpu_cmdq *cmdq, struct ivpu_job *job)
181 {
182         struct ivpu_device *vdev = job->vdev;
183         struct vpu_job_queue_header *header = &cmdq->jobq->header;
184         struct vpu_job_queue_entry *entry;
185         u32 tail = READ_ONCE(header->tail);
186         u32 next_entry = (tail + 1) % cmdq->entry_count;
187
188         /* Check if there is space left in job queue */
189         if (next_entry == header->head) {
190                 ivpu_dbg(vdev, JOB, "Job queue full: ctx %d engine %d db %d head %d tail %d\n",
191                          job->file_priv->ctx.id, job->engine_idx, cmdq->db_id, header->head, tail);
192                 return -EBUSY;
193         }
194
195         entry = &cmdq->jobq->job[tail];
196         entry->batch_buf_addr = job->cmd_buf_vpu_addr;
197         entry->job_id = job->job_id;
198         entry->flags = 0;
199         wmb(); /* Ensure that tail is updated after filling entry */
200         header->tail = next_entry;
201         wmb(); /* Flush WC buffer for jobq header */
202
203         return 0;
204 }
205
206 struct ivpu_fence {
207         struct dma_fence base;
208         spinlock_t lock; /* protects base */
209         struct ivpu_device *vdev;
210 };
211
212 static inline struct ivpu_fence *to_vpu_fence(struct dma_fence *fence)
213 {
214         return container_of(fence, struct ivpu_fence, base);
215 }
216
217 static const char *ivpu_fence_get_driver_name(struct dma_fence *fence)
218 {
219         return DRIVER_NAME;
220 }
221
222 static const char *ivpu_fence_get_timeline_name(struct dma_fence *fence)
223 {
224         struct ivpu_fence *ivpu_fence = to_vpu_fence(fence);
225
226         return dev_name(ivpu_fence->vdev->drm.dev);
227 }
228
229 static const struct dma_fence_ops ivpu_fence_ops = {
230         .get_driver_name = ivpu_fence_get_driver_name,
231         .get_timeline_name = ivpu_fence_get_timeline_name,
232 };
233
234 static struct dma_fence *ivpu_fence_create(struct ivpu_device *vdev)
235 {
236         struct ivpu_fence *fence;
237
238         fence = kzalloc(sizeof(*fence), GFP_KERNEL);
239         if (!fence)
240                 return NULL;
241
242         fence->vdev = vdev;
243         spin_lock_init(&fence->lock);
244         dma_fence_init(&fence->base, &ivpu_fence_ops, &fence->lock, dma_fence_context_alloc(1), 1);
245
246         return &fence->base;
247 }
248
249 static void job_get(struct ivpu_job *job, struct ivpu_job **link)
250 {
251         struct ivpu_device *vdev = job->vdev;
252
253         kref_get(&job->ref);
254         *link = job;
255
256         ivpu_dbg(vdev, KREF, "Job get: id %u refcount %u\n", job->job_id, kref_read(&job->ref));
257 }
258
259 static void job_release(struct kref *ref)
260 {
261         struct ivpu_job *job = container_of(ref, struct ivpu_job, ref);
262         struct ivpu_device *vdev = job->vdev;
263         u32 i;
264
265         for (i = 0; i < job->bo_count; i++)
266                 if (job->bos[i])
267                         drm_gem_object_put(&job->bos[i]->base);
268
269         dma_fence_put(job->done_fence);
270         ivpu_file_priv_put(&job->file_priv);
271
272         ivpu_dbg(vdev, KREF, "Job released: id %u\n", job->job_id);
273         kfree(job);
274
275         /* Allow the VPU to get suspended, must be called after ivpu_file_priv_put() */
276         ivpu_rpm_put(vdev);
277 }
278
279 static void job_put(struct ivpu_job *job)
280 {
281         struct ivpu_device *vdev = job->vdev;
282
283         ivpu_dbg(vdev, KREF, "Job put: id %u refcount %u\n", job->job_id, kref_read(&job->ref));
284         kref_put(&job->ref, job_release);
285 }
286
287 static struct ivpu_job *
288 ivpu_create_job(struct ivpu_file_priv *file_priv, u32 engine_idx, u32 bo_count)
289 {
290         struct ivpu_device *vdev = file_priv->vdev;
291         struct ivpu_job *job;
292         size_t buf_size;
293         int ret;
294
295         ret = ivpu_rpm_get(vdev);
296         if (ret < 0)
297                 return NULL;
298
299         buf_size = sizeof(*job) + bo_count * sizeof(struct ivpu_bo *);
300         job = kzalloc(buf_size, GFP_KERNEL);
301         if (!job)
302                 goto err_rpm_put;
303
304         kref_init(&job->ref);
305
306         job->vdev = vdev;
307         job->engine_idx = engine_idx;
308         job->bo_count = bo_count;
309         job->done_fence = ivpu_fence_create(vdev);
310         if (!job->done_fence) {
311                 ivpu_warn_ratelimited(vdev, "Failed to create a fence\n");
312                 goto err_free_job;
313         }
314
315         job->file_priv = ivpu_file_priv_get(file_priv);
316
317         ivpu_dbg(vdev, JOB, "Job created: ctx %2d engine %d", file_priv->ctx.id, job->engine_idx);
318
319         return job;
320
321 err_free_job:
322         kfree(job);
323 err_rpm_put:
324         ivpu_rpm_put(vdev);
325         return NULL;
326 }
327
328 static int ivpu_job_done(struct ivpu_device *vdev, u32 job_id, u32 job_status)
329 {
330         struct ivpu_job *job;
331
332         job = xa_erase(&vdev->submitted_jobs_xa, job_id);
333         if (!job)
334                 return -ENOENT;
335
336         if (job->file_priv->has_mmu_faults)
337                 job_status = VPU_JSM_STATUS_ABORTED;
338
339         job->bos[CMD_BUF_IDX]->job_status = job_status;
340         dma_fence_signal(job->done_fence);
341
342         ivpu_dbg(vdev, JOB, "Job complete:  id %3u ctx %2d engine %d status 0x%x\n",
343                  job->job_id, job->file_priv->ctx.id, job->engine_idx, job_status);
344
345         job_put(job);
346         return 0;
347 }
348
349 static void ivpu_job_done_message(struct ivpu_device *vdev, void *msg)
350 {
351         struct vpu_ipc_msg_payload_job_done *payload;
352         struct vpu_jsm_msg *job_ret_msg = msg;
353         int ret;
354
355         payload = (struct vpu_ipc_msg_payload_job_done *)&job_ret_msg->payload;
356
357         ret = ivpu_job_done(vdev, payload->job_id, payload->job_status);
358         if (ret)
359                 ivpu_err(vdev, "Failed to finish job %d: %d\n", payload->job_id, ret);
360 }
361
362 void ivpu_jobs_abort_all(struct ivpu_device *vdev)
363 {
364         struct ivpu_job *job;
365         unsigned long id;
366
367         xa_for_each(&vdev->submitted_jobs_xa, id, job)
368                 ivpu_job_done(vdev, id, VPU_JSM_STATUS_ABORTED);
369 }
370
371 static int ivpu_direct_job_submission(struct ivpu_job *job)
372 {
373         struct ivpu_file_priv *file_priv = job->file_priv;
374         struct ivpu_device *vdev = job->vdev;
375         struct xa_limit job_id_range;
376         struct ivpu_cmdq *cmdq;
377         int ret;
378
379         mutex_lock(&file_priv->lock);
380
381         cmdq = ivpu_cmdq_acquire(job->file_priv, job->engine_idx);
382         if (!cmdq) {
383                 ivpu_warn(vdev, "Failed get job queue, ctx %d engine %d\n",
384                           file_priv->ctx.id, job->engine_idx);
385                 ret = -EINVAL;
386                 goto err_unlock;
387         }
388
389         job_id_range.min = FIELD_PREP(JOB_ID_CONTEXT_MASK, (file_priv->ctx.id - 1));
390         job_id_range.max = job_id_range.min | JOB_ID_JOB_MASK;
391
392         job_get(job, &job);
393         ret = xa_alloc(&vdev->submitted_jobs_xa, &job->job_id, job, job_id_range, GFP_KERNEL);
394         if (ret) {
395                 ivpu_warn_ratelimited(vdev, "Failed to allocate job id: %d\n", ret);
396                 goto err_job_put;
397         }
398
399         ret = ivpu_cmdq_push_job(cmdq, job);
400         if (ret)
401                 goto err_xa_erase;
402
403         ivpu_dbg(vdev, JOB, "Job submitted: id %3u addr 0x%llx ctx %2d engine %d next %d\n",
404                  job->job_id, job->cmd_buf_vpu_addr, file_priv->ctx.id,
405                  job->engine_idx, cmdq->jobq->header.tail);
406
407         if (ivpu_test_mode == IVPU_TEST_MODE_NULL_HW) {
408                 ivpu_job_done(vdev, job->job_id, VPU_JSM_STATUS_SUCCESS);
409                 cmdq->jobq->header.head = cmdq->jobq->header.tail;
410                 wmb(); /* Flush WC buffer for jobq header */
411         } else {
412                 ivpu_cmdq_ring_db(vdev, cmdq);
413         }
414
415         mutex_unlock(&file_priv->lock);
416         return 0;
417
418 err_xa_erase:
419         xa_erase(&vdev->submitted_jobs_xa, job->job_id);
420 err_job_put:
421         job_put(job);
422 err_unlock:
423         mutex_unlock(&file_priv->lock);
424         return ret;
425 }
426
427 static int
428 ivpu_job_prepare_bos_for_submit(struct drm_file *file, struct ivpu_job *job, u32 *buf_handles,
429                                 u32 buf_count, u32 commands_offset)
430 {
431         struct ivpu_file_priv *file_priv = file->driver_priv;
432         struct ivpu_device *vdev = file_priv->vdev;
433         struct ww_acquire_ctx acquire_ctx;
434         struct ivpu_bo *bo;
435         int ret;
436         u32 i;
437
438         for (i = 0; i < buf_count; i++) {
439                 struct drm_gem_object *obj = drm_gem_object_lookup(file, buf_handles[i]);
440
441                 if (!obj)
442                         return -ENOENT;
443
444                 job->bos[i] = to_ivpu_bo(obj);
445
446                 ret = ivpu_bo_pin(job->bos[i]);
447                 if (ret)
448                         return ret;
449         }
450
451         bo = job->bos[CMD_BUF_IDX];
452         if (!dma_resv_test_signaled(bo->base.resv, DMA_RESV_USAGE_READ)) {
453                 ivpu_warn(vdev, "Buffer is already in use\n");
454                 return -EBUSY;
455         }
456
457         if (commands_offset >= bo->base.size) {
458                 ivpu_warn(vdev, "Invalid command buffer offset %u\n", commands_offset);
459                 return -EINVAL;
460         }
461
462         job->cmd_buf_vpu_addr = bo->vpu_addr + commands_offset;
463
464         ret = drm_gem_lock_reservations((struct drm_gem_object **)job->bos, buf_count,
465                                         &acquire_ctx);
466         if (ret) {
467                 ivpu_warn(vdev, "Failed to lock reservations: %d\n", ret);
468                 return ret;
469         }
470
471         for (i = 0; i < buf_count; i++) {
472                 ret = dma_resv_reserve_fences(job->bos[i]->base.resv, 1);
473                 if (ret) {
474                         ivpu_warn(vdev, "Failed to reserve fences: %d\n", ret);
475                         goto unlock_reservations;
476                 }
477         }
478
479         for (i = 0; i < buf_count; i++)
480                 dma_resv_add_fence(job->bos[i]->base.resv, job->done_fence, DMA_RESV_USAGE_WRITE);
481
482 unlock_reservations:
483         drm_gem_unlock_reservations((struct drm_gem_object **)job->bos, buf_count, &acquire_ctx);
484
485         wmb(); /* Flush write combining buffers */
486
487         return ret;
488 }
489
490 int ivpu_submit_ioctl(struct drm_device *dev, void *data, struct drm_file *file)
491 {
492         struct ivpu_file_priv *file_priv = file->driver_priv;
493         struct ivpu_device *vdev = file_priv->vdev;
494         struct drm_ivpu_submit *params = data;
495         struct ivpu_job *job;
496         u32 *buf_handles;
497         int idx, ret;
498
499         if (params->engine > DRM_IVPU_ENGINE_COPY)
500                 return -EINVAL;
501
502         if (params->buffer_count == 0 || params->buffer_count > JOB_MAX_BUFFER_COUNT)
503                 return -EINVAL;
504
505         if (!IS_ALIGNED(params->commands_offset, 8))
506                 return -EINVAL;
507
508         if (!file_priv->ctx.id)
509                 return -EINVAL;
510
511         if (file_priv->has_mmu_faults)
512                 return -EBADFD;
513
514         buf_handles = kcalloc(params->buffer_count, sizeof(u32), GFP_KERNEL);
515         if (!buf_handles)
516                 return -ENOMEM;
517
518         ret = copy_from_user(buf_handles,
519                              (void __user *)params->buffers_ptr,
520                              params->buffer_count * sizeof(u32));
521         if (ret) {
522                 ret = -EFAULT;
523                 goto free_handles;
524         }
525
526         if (!drm_dev_enter(&vdev->drm, &idx)) {
527                 ret = -ENODEV;
528                 goto free_handles;
529         }
530
531         ivpu_dbg(vdev, JOB, "Submit ioctl: ctx %u buf_count %u\n",
532                  file_priv->ctx.id, params->buffer_count);
533
534         job = ivpu_create_job(file_priv, params->engine, params->buffer_count);
535         if (!job) {
536                 ivpu_err(vdev, "Failed to create job\n");
537                 ret = -ENOMEM;
538                 goto dev_exit;
539         }
540
541         ret = ivpu_job_prepare_bos_for_submit(file, job, buf_handles, params->buffer_count,
542                                               params->commands_offset);
543         if (ret) {
544                 ivpu_err(vdev, "Failed to prepare job, ret %d\n", ret);
545                 goto job_put;
546         }
547
548         ret = ivpu_direct_job_submission(job);
549         if (ret) {
550                 dma_fence_signal(job->done_fence);
551                 ivpu_err(vdev, "Failed to submit job to the HW, ret %d\n", ret);
552         }
553
554 job_put:
555         job_put(job);
556 dev_exit:
557         drm_dev_exit(idx);
558 free_handles:
559         kfree(buf_handles);
560
561         return ret;
562 }
563
564 static int ivpu_job_done_thread(void *arg)
565 {
566         struct ivpu_device *vdev = (struct ivpu_device *)arg;
567         struct ivpu_ipc_consumer cons;
568         struct vpu_jsm_msg jsm_msg;
569         bool jobs_submitted;
570         unsigned int timeout;
571         int ret;
572
573         ivpu_dbg(vdev, JOB, "Started %s\n", __func__);
574
575         ivpu_ipc_consumer_add(vdev, &cons, VPU_IPC_CHAN_JOB_RET);
576
577         while (!kthread_should_stop()) {
578                 timeout = ivpu_tdr_timeout_ms ? ivpu_tdr_timeout_ms : vdev->timeout.tdr;
579                 jobs_submitted = !xa_empty(&vdev->submitted_jobs_xa);
580                 ret = ivpu_ipc_receive(vdev, &cons, NULL, &jsm_msg, timeout);
581                 if (!ret) {
582                         ivpu_job_done_message(vdev, &jsm_msg);
583                 } else if (ret == -ETIMEDOUT) {
584                         if (jobs_submitted && !xa_empty(&vdev->submitted_jobs_xa)) {
585                                 ivpu_err(vdev, "TDR detected, timeout %d ms", timeout);
586                                 ivpu_hw_diagnose_failure(vdev);
587                                 ivpu_pm_schedule_recovery(vdev);
588                         }
589                 }
590         }
591
592         ivpu_ipc_consumer_del(vdev, &cons);
593
594         ivpu_jobs_abort_all(vdev);
595
596         ivpu_dbg(vdev, JOB, "Stopped %s\n", __func__);
597         return 0;
598 }
599
600 int ivpu_job_done_thread_init(struct ivpu_device *vdev)
601 {
602         struct task_struct *thread;
603
604         thread = kthread_run(&ivpu_job_done_thread, (void *)vdev, "ivpu_job_done_thread");
605         if (IS_ERR(thread)) {
606                 ivpu_err(vdev, "Failed to start job completion thread\n");
607                 return -EIO;
608         }
609
610         get_task_struct(thread);
611         wake_up_process(thread);
612
613         vdev->job_done_thread = thread;
614
615         return 0;
616 }
617
618 void ivpu_job_done_thread_fini(struct ivpu_device *vdev)
619 {
620         kthread_stop(vdev->job_done_thread);
621         put_task_struct(vdev->job_done_thread);
622 }