1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 43284ba5..76413600 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -629,6 +629,10 @@ struct gpu_ops {
                 u32 (*get_preempt_timeout)(struct gk20a *g);
                 void (*post_event_id)(struct tsg_gk20a *tsg, int event_id);
                 void (*ch_abort_clean_up)(struct channel_gk20a *ch);
+                bool (*check_tsg_ctxsw_timeout)(struct tsg_gk20a *tsg,
+                                                bool *verbose, u32 *ms);
+                bool (*check_ch_ctxsw_timeout)(struct channel_gk20a *ch,
+                                                bool *verbose, u32 *ms);
 #ifdef CONFIG_TEGRA_GK20A_NVHOST
                 int (*alloc_syncpt_buf)(struct channel_gk20a *c,
                                 u32 syncpt_id, struct nvgpu_mem *syncpt_buf);