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 | e61235d | 2015-08-25 11:05:36 +0200 | [diff] [blame] | 30 | static struct fence *amdgpu_sched_dependency(struct amd_sched_job *job) |
| 31 | { |
| 32 | struct amdgpu_job *sched_job = (struct amdgpu_job *)job; |
| 33 | return amdgpu_sync_get_fence(&sched_job->ibs->sync); |
| 34 | } |
| 35 | |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame] | 36 | static struct fence *amdgpu_sched_run_job(struct amd_sched_job *job) |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 37 | { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 38 | struct amdgpu_job *sched_job; |
Chunming Zhou | 7484667 | 2015-08-04 11:30:09 +0800 | [diff] [blame] | 39 | struct amdgpu_fence *fence; |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame] | 40 | int r; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 41 | |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 42 | if (!job) { |
Chunming Zhou | 4cef926 | 2015-08-05 19:52:14 +0800 | [diff] [blame] | 43 | DRM_ERROR("job is null\n"); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 44 | return NULL; |
Chunming Zhou | 4cef926 | 2015-08-05 19:52:14 +0800 | [diff] [blame] | 45 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 46 | sched_job = (struct amdgpu_job *)job; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 47 | mutex_lock(&sched_job->job_lock); |
| 48 | r = amdgpu_ib_schedule(sched_job->adev, |
| 49 | sched_job->num_ibs, |
| 50 | sched_job->ibs, |
Chunming Zhou | 84f76ea | 2015-08-24 12:47:36 +0800 | [diff] [blame] | 51 | sched_job->base.owner); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 52 | if (r) |
| 53 | goto err; |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 54 | 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] | 55 | |
Christian König | bf7ebae | 2015-08-18 15:30:26 +0200 | [diff] [blame] | 56 | if (sched_job->free_job) |
| 57 | sched_job->free_job(sched_job); |
| 58 | |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 59 | mutex_unlock(&sched_job->job_lock); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 60 | return &fence->base; |
| 61 | |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 62 | err: |
| 63 | DRM_ERROR("Run job error\n"); |
| 64 | mutex_unlock(&sched_job->job_lock); |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame] | 65 | job->sched->ops->process_job(job); |
Christian König | 6f0e54a | 2015-08-05 21:22:10 +0200 | [diff] [blame] | 66 | return NULL; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 67 | } |
| 68 | |
Christian König | bd755d0 | 2015-08-24 14:57:26 +0200 | [diff] [blame] | 69 | static void amdgpu_sched_process_job(struct amd_sched_job *job) |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 70 | { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 71 | struct amdgpu_job *sched_job; |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 72 | |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 73 | if (!job) { |
Chunming Zhou | 953e8fd | 2015-08-06 15:19:12 +0800 | [diff] [blame] | 74 | DRM_ERROR("job is null\n"); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 75 | return; |
Chunming Zhou | 953e8fd | 2015-08-06 15:19:12 +0800 | [diff] [blame] | 76 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 77 | sched_job = (struct amdgpu_job *)job; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 78 | /* after processing job, free memory */ |
| 79 | fence_put(&sched_job->base.s_fence->base); |
| 80 | kfree(sched_job); |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 81 | } |
| 82 | |
| 83 | struct amd_sched_backend_ops amdgpu_sched_ops = { |
Christian König | e61235d | 2015-08-25 11:05:36 +0200 | [diff] [blame] | 84 | .dependency = amdgpu_sched_dependency, |
Chunming Zhou | c1b69ed | 2015-07-21 13:45:14 +0800 | [diff] [blame] | 85 | .run_job = amdgpu_sched_run_job, |
| 86 | .process_job = amdgpu_sched_process_job |
| 87 | }; |
| 88 | |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 89 | int amdgpu_sched_ib_submit_kernel_helper(struct amdgpu_device *adev, |
| 90 | struct amdgpu_ring *ring, |
| 91 | struct amdgpu_ib *ibs, |
| 92 | unsigned num_ibs, |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 93 | int (*free_job)(struct amdgpu_job *), |
Chunming Zhou | 1763552 | 2015-08-03 11:43:19 +0800 | [diff] [blame] | 94 | void *owner, |
| 95 | struct fence **f) |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 96 | { |
| 97 | int r = 0; |
| 98 | if (amdgpu_enable_scheduler) { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 99 | struct amdgpu_job *job = |
| 100 | kzalloc(sizeof(struct amdgpu_job), GFP_KERNEL); |
| 101 | if (!job) |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 102 | return -ENOMEM; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 103 | job->base.sched = ring->scheduler; |
| 104 | job->base.s_entity = &adev->kernel_ctx.rings[ring->idx].entity; |
| 105 | job->adev = adev; |
| 106 | job->ibs = ibs; |
| 107 | job->num_ibs = num_ibs; |
Chunming Zhou | 84f76ea | 2015-08-24 12:47:36 +0800 | [diff] [blame] | 108 | job->base.owner = owner; |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 109 | mutex_init(&job->job_lock); |
| 110 | job->free_job = free_job; |
| 111 | mutex_lock(&job->job_lock); |
Christian König | 6c85927 | 2015-08-20 16:12:50 +0200 | [diff] [blame] | 112 | r = amd_sched_entity_push_job((struct amd_sched_job *)job); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 113 | if (r) { |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 114 | mutex_unlock(&job->job_lock); |
| 115 | kfree(job); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 116 | return r; |
| 117 | } |
Chunming Zhou | bb977d3 | 2015-08-18 15:16:40 +0800 | [diff] [blame] | 118 | *f = fence_get(&job->base.s_fence->base); |
| 119 | mutex_unlock(&job->job_lock); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 120 | } else { |
Chunming Zhou | 4af9f07 | 2015-08-03 12:57:31 +0800 | [diff] [blame] | 121 | r = amdgpu_ib_schedule(adev, num_ibs, ibs, owner); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 122 | if (r) |
| 123 | return r; |
Chunming Zhou | 281b422 | 2015-08-12 12:58:31 +0800 | [diff] [blame] | 124 | *f = fence_get(&ibs[num_ibs - 1].fence->base); |
Chunming Zhou | f556cb0c | 2015-08-02 11:18:04 +0800 | [diff] [blame] | 125 | } |
Chunming Zhou | 3c62338 | 2015-08-20 18:33:59 +0800 | [diff] [blame] | 126 | |
Chunming Zhou | 1763552 | 2015-08-03 11:43:19 +0800 | [diff] [blame] | 127 | return 0; |
Chunming Zhou | 3c704e9 | 2015-07-29 10:33:14 +0800 | [diff] [blame] | 128 | } |