1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
index 0424e74d..f6aabd82 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
@@ -646,7 +646,7 @@ struct gpu_ops {
                                 u32 flags);
                 int (*resetup_ramfc)(struct channel_gk20a *c);
                 int (*preempt_channel)(struct gk20a *g, u32 chid);
-                int (*preempt_tsg)(struct gk20a *g, u32 tsgid);
+                int (*preempt_tsg)(struct gk20a *g, struct tsg_gk20a *tsg);
                 int (*enable_tsg)(struct tsg_gk20a *tsg);
                 int (*disable_tsg)(struct tsg_gk20a *tsg);
                 int (*tsg_verify_channel_status)(struct channel_gk20a *ch);