#include <drm/drmP.h>
 #include "amdgpu.h"
 
-static struct fence *amdgpu_sched_run_job(struct amd_gpu_scheduler *sched,
-                                         struct amd_sched_entity *entity,
-                                         struct amd_sched_job *job)
+static struct fence *amdgpu_sched_run_job(struct amd_sched_job *job)
 {
-       int r = 0;
        struct amdgpu_job *sched_job;
        struct amdgpu_fence *fence;
+       int r;
 
        if (!job) {
                DRM_ERROR("job is null\n");
 err:
        DRM_ERROR("Run job error\n");
        mutex_unlock(&sched_job->job_lock);
-       sched->ops->process_job(sched, (struct amd_sched_job *)sched_job);
+       job->sched->ops->process_job(job);
        return NULL;
 }
 
-static void amdgpu_sched_process_job(struct amd_gpu_scheduler *sched,
-                                    struct amd_sched_job *job)
+static void amdgpu_sched_process_job(struct amd_sched_job *job)
 {
        struct amdgpu_job *sched_job;
 
 
        amd_sched_fence_signal(sched_job->s_fence);
        atomic_dec(&sched->hw_rq_count);
        fence_put(&sched_job->s_fence->base);
-       sched->ops->process_job(sched, sched_job);
+       sched->ops->process_job(sched_job);
        wake_up_interruptible(&sched->wait_queue);
 }
 
                        continue;
                atomic_inc(&sched->hw_rq_count);
 
-               fence = sched->ops->run_job(sched, c_entity, job);
+               fence = sched->ops->run_job(job);
                if (fence) {
                        r = fence_add_callback(fence, &job->cb,
                                               amd_sched_process_job);
 
  * these functions should be implemented in driver side
 */
 struct amd_sched_backend_ops {
-       struct fence *(*run_job)(struct amd_gpu_scheduler *sched,
-                                struct amd_sched_entity *c_entity,
-                                struct amd_sched_job *job);
-       void (*process_job)(struct amd_gpu_scheduler *sched,
-                                   struct amd_sched_job *job);
+       struct fence *(*run_job)(struct amd_sched_job *job);
+       void (*process_job)(struct amd_sched_job *job);
 };
 
 /**