2 files changed, 23 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/hw_therm_gp10b.h b/drivers/gpu/nvgpu/gp10b/hw_therm_gp10b.h
index 17b1fb7e..8a587b7c 100644
--- a/drivers/gpu/nvgpu/gp10b/hw_therm_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/hw_therm_gp10b.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2014-2015, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -246,6 +246,14 @@ static inline u32 therm_gate_ctrl_eng_idle_filt_mant_m(void)
 {
         return 0x7 << 13;
 }
+static inline u32 therm_gate_ctrl_eng_delay_before_f(u32 v)
+{
+        return (v & 0xf) << 16;
+}
+static inline u32 therm_gate_ctrl_eng_delay_before_m(void)
+{
+        return 0xf << 16;
+}
 static inline u32 therm_gate_ctrl_eng_delay_after_f(u32 v)
 {
         return (v & 0xf) << 20;
diff --git a/drivers/gpu/nvgpu/gp10b/therm_gp10b.c b/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
index 7bf0c0e5..687593b0 100644
--- a/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
@@ -78,8 +78,22 @@ static int gp10b_init_therm_setup_hw(struct gk20a *g)
         return 0;
 }
 
+static int gp10b_update_therm_gate_ctrl(struct gk20a *g)
+{
+        u32 gate_ctrl;
+
+        gate_ctrl = gk20a_readl(g, therm_gate_ctrl_r(ENGINE_CE2_GK20A));
+        gate_ctrl = set_field(gate_ctrl,
+                therm_gate_ctrl_eng_delay_before_m(),
+                therm_gate_ctrl_eng_delay_before_f(4));
+        gk20a_writel(g, therm_gate_ctrl_r(ENGINE_CE2_GK20A), gate_ctrl);
+
+        return 0;
+}
+
 void gp10b_init_therm_ops(struct gpu_ops *gops)
 {
         gops->therm.init_therm_setup_hw = gp10b_init_therm_setup_hw;
+        gops->therm.update_therm_gate_ctrl = gp10b_update_therm_gate_ctrl;
 
 }