Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 Advanced Micro Devices, Inc. |
| 3 | * |
| 4 | * Permission is hereby granted, free of charge, to any person obtaining a |
| 5 | * copy of this software and associated documentation files (the "Software"), |
| 6 | * to deal in the Software without restriction, including without limitation |
| 7 | * the rights to use, copy, modify, merge, publish, distribute, sublicense, |
| 8 | * and/or sell copies of the Software, and to permit persons to whom the |
| 9 | * Software is furnished to do so, subject to the following conditions: |
| 10 | * |
| 11 | * The above copyright notice and this permission notice shall be included in |
| 12 | * all copies or substantial portions of the Software. |
| 13 | * |
| 14 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 15 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 16 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
| 17 | * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR |
| 18 | * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, |
| 19 | * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR |
| 20 | * OTHER DEALINGS IN THE SOFTWARE. |
| 21 | * |
| 22 | * |
| 23 | */ |
| 24 | #include <linux/kthread.h> |
| 25 | #include <linux/wait.h> |
| 26 | #include <linux/sched.h> |
| 27 | #include <drm/drmP.h> |
| 28 | #include "amdgpu.h" |
| 29 | |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame^] | 30 | static struct fence *amdgpu_sched_run_job(struct amd_sched_job *job) |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 31 | { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 32 | struct amdgpu_job *sched_job; |
Chunming Zhou | 7484667 | 2015-08-04 11:30:09 +0800 | [diff] [blame] | 33 | struct amdgpu_fence *fence; |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame^] | 34 | int r; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 35 | |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 36 | if (!job) { |
Chunming Zhou | 4cef926 | 2015-08-05 19:52:14 +0800 | [diff] [blame] | 37 | DRM_ERROR("job is null\n"); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 38 | return NULL; |
Chunming Zhou | 4cef926 | 2015-08-05 19:52:14 +0800 | [diff] [blame] | 39 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 40 | sched_job = (struct amdgpu_job *)job; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 41 | mutex_lock(&sched_job->job_lock); |
| 42 | r = amdgpu_ib_schedule(sched_job->adev, |
| 43 | sched_job->num_ibs, |
| 44 | sched_job->ibs, |
Chunming Zhou | 84f76ea | 2015-08-24 12:47:36 +0800 | [diff] [blame] | 45 | sched_job->base.owner); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 46 | if (r) |
| 47 | goto err; |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 48 | fence = amdgpu_fence_ref(sched_job->ibs[sched_job->num_ibs - 1].fence); |
Chunming Zhou | 7484667 | 2015-08-04 11:30:09 +0800 | [diff] [blame] | 49 | |
Christian König | bf7ebae | 2015-08-18 15:30:26 +0200 | [diff] [blame] | 50 | if (sched_job->free_job) |
| 51 | sched_job->free_job(sched_job); |
| 52 | |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 53 | mutex_unlock(&sched_job->job_lock); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 54 | return &fence->base; |
| 55 | |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 56 | err: |
| 57 | DRM_ERROR("Run job error\n"); |
| 58 | mutex_unlock(&sched_job->job_lock); |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame^] | 59 | job->sched->ops->process_job(job); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 60 | return NULL; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 61 | } |
| 62 | |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame^] | 63 | static void amdgpu_sched_process_job(struct amd_sched_job *job) |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 64 | { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 65 | struct amdgpu_job *sched_job; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 66 | |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 67 | if (!job) { |
Chunming Zhou | 953e8fd | 2015-08-06 15:19:12 +0800 | [diff] [blame] | 68 | DRM_ERROR("job is null\n"); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 69 | return; |
Chunming Zhou | 953e8fd | 2015-08-06 15:19:12 +0800 | [diff] [blame] | 70 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 71 | sched_job = (struct amdgpu_job *)job; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 72 | /* after processing job, free memory */ |
| 73 | fence_put(&sched_job->base.s_fence->base); |
| 74 | kfree(sched_job); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 75 | } |
| 76 | |
| 77 | struct amd_sched_backend_ops amdgpu_sched_ops = { |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 78 | .run_job = amdgpu_sched_run_job, |
| 79 | .process_job = amdgpu_sched_process_job |
| 80 | }; |
| 81 | |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 82 | int amdgpu_sched_ib_submit_kernel_helper(struct amdgpu_device *adev, |
| 83 | struct amdgpu_ring *ring, |
| 84 | struct amdgpu_ib *ibs, |
| 85 | unsigned num_ibs, |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 86 | int (*free_job)(struct amdgpu_job *), |
Chunming Zhou | 1763552 | 2015-08-03 11:43:19 +0800 | [diff] [blame] | 87 | void *owner, |
| 88 | struct fence **f) |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 89 | { |
| 90 | int r = 0; |
| 91 | if (amdgpu_enable_scheduler) { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 92 | struct amdgpu_job *job = |
| 93 | kzalloc(sizeof(struct amdgpu_job), GFP_KERNEL); |
| 94 | if (!job) |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 95 | return -ENOMEM; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 96 | job->base.sched = ring->scheduler; |
| 97 | job->base.s_entity = &adev->kernel_ctx.rings[ring->idx].entity; |
| 98 | job->adev = adev; |
| 99 | job->ibs = ibs; |
| 100 | job->num_ibs = num_ibs; |
Chunming Zhou | 84f76ea | 2015-08-24 12:47:36 +0800 | [diff] [blame] | 101 | job->base.owner = owner; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 102 | mutex_init(&job->job_lock); |
| 103 | job->free_job = free_job; |
| 104 | mutex_lock(&job->job_lock); |
Christian König | 6c85927 | 2015-08-20 16:12:50 +0200 | [diff] [blame] | 105 | r = amd_sched_entity_push_job((struct amd_sched_job *)job); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 106 | if (r) { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 107 | mutex_unlock(&job->job_lock); |
| 108 | kfree(job); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 109 | return r; |
| 110 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 111 | *f = fence_get(&job->base.s_fence->base); |
| 112 | mutex_unlock(&job->job_lock); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 113 | } else { |
Chunming Zhou | 4af9f07 | 2015-08-03 12:57:31 +0800 | [diff] [blame] | 114 | r = amdgpu_ib_schedule(adev, num_ibs, ibs, owner); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 115 | if (r) |
| 116 | return r; |
Chunming Zhou | 281b422 | 2015-08-12 12:58:31 +0800 | [diff] [blame] | 117 | *f = fence_get(&ibs[num_ibs - 1].fence->base); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 118 | } |
Chunming Zhou | 3c62338 | 2015-08-20 18:33:59 +0800 | [diff] [blame] | 119 | |
Chunming Zhou | 1763552 | 2015-08-03 11:43:19 +0800 | [diff] [blame] | 120 | return 0; |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 121 | } |