1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
index 1942b1e7..7d9f5f47 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.c
@@ -284,6 +284,11 @@ static int gr_gp10b_calc_global_ctx_buffer_size(struct gk20a *g)
         return size;
 }
 
+static void gr_gp10b_set_go_idle_timeout(struct gk20a *g, u32 data)
+{
+        gk20a_writel(g, gr_fe_go_idle_timeout_r(), data);
+}
+
 static int gr_gp10b_handle_sw_method(struct gk20a *g, u32 addr,
                                      u32 class_num, u32 offset, u32 data)
 {
@@ -310,6 +315,9 @@ static int gr_gp10b_handle_sw_method(struct gk20a *g, u32 addr,
                 case NVC097_SET_ALPHA_CIRCULAR_BUFFER_SIZE:
                         g->ops.gr.set_alpha_circular_buffer_size(g, data);
                         break;
+                case NVC097_SET_GO_IDLE_TIMEOUT:
+                        gr_gp10b_set_go_idle_timeout(g, data);
+                        break;
                 default:
                         goto fail;
                 }