1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 17f662df..45fa58f1 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -566,7 +566,8 @@ struct gpu_ops {
                 int (*tsg_verify_channel_status)(struct channel_gk20a *ch);
                 void (*tsg_verify_status_ctx_reload)(struct channel_gk20a *ch);
                 void (*tsg_verify_status_faulted)(struct channel_gk20a *ch);
-                int (*reschedule_runlist)(struct gk20a *g, u32 runlist_id);
+                int (*reschedule_runlist)(struct channel_gk20a *ch,
+                                bool preempt_next);
                 int (*update_runlist)(struct gk20a *g, u32 runlist_id,
                                 u32 chid, bool add,
                                 bool wait_for_finish);