From: Chunming Zhou <david1.z...@amd.com>

Signed-off-by: Chunming Zhou <david1.zhou at amd.com>
Reviewed-by: Christian K?nig <christian.koenig at amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu.h       |  7 +++++++
 drivers/gpu/drm/amd/amdgpu/amdgpu_sched.c | 35 +++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu.h 
b/drivers/gpu/drm/amd/amdgpu/amdgpu.h
index 3dfff89..7fecb44 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu.h
@@ -863,6 +863,13 @@ enum amdgpu_ring_type {

 extern struct amd_sched_backend_ops amdgpu_sched_ops;

+int amdgpu_sched_ib_submit_kernel_helper(struct amdgpu_device *adev,
+                                        struct amdgpu_ring *ring,
+                                        struct amdgpu_ib *ibs,
+                                        unsigned num_ibs,
+                                        int (*free_job)(struct 
amdgpu_cs_parser *),
+                                        void *owner);
+
 struct amdgpu_ring {
        struct amdgpu_device            *adev;
        const struct amdgpu_ring_funcs  *funcs;
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_sched.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_sched.c
index b913c22..d682fab 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_sched.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_sched.c
@@ -108,3 +108,38 @@ struct amd_sched_backend_ops amdgpu_sched_ops = {
        .process_job = amdgpu_sched_process_job
 };

+int amdgpu_sched_ib_submit_kernel_helper(struct amdgpu_device *adev,
+                                        struct amdgpu_ring *ring,
+                                        struct amdgpu_ib *ibs,
+                                        unsigned num_ibs,
+                                        int (*free_job)(struct 
amdgpu_cs_parser *),
+                                        void *owner)
+{
+       int r = 0;
+       if (amdgpu_enable_scheduler) {
+               uint64_t v_seq;
+               struct amdgpu_cs_parser *sched_job =
+                       amdgpu_cs_parser_create(adev,
+                                               owner,
+                                               adev->kernel_ctx,
+                                               ibs, 1);
+               if(!sched_job) {
+                       return -ENOMEM;
+               }
+               sched_job->free_job = free_job;
+               v_seq = 
atomic64_inc_return(&adev->kernel_ctx->rings[ring->idx].c_entity.last_queued_v_seq);
+               ibs[num_ibs - 1].sequence = v_seq;
+               amd_sched_push_job(ring->scheduler,
+                                  &adev->kernel_ctx->rings[ring->idx].c_entity,
+                                  sched_job);
+               r = amd_sched_wait_emit(
+                       &adev->kernel_ctx->rings[ring->idx].c_entity,
+                       v_seq,
+                       false,
+                       -1);
+               if (r)
+                       WARN(true, "emit timeout\n");
+       } else
+               r = amdgpu_ib_schedule(adev, 1, ibs, owner);
+       return r;
+}
-- 
1.8.3.1

Reply via email to