1 files changed, 6 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/therm_gp10b.c b/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
index 5763b3b1..63efc945 100644
--- a/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/therm_gp10b.c
@@ -82,13 +82,16 @@ static int gp10b_update_therm_gate_ctrl(struct gk20a *g)
 {
         u32 gate_ctrl;
         u32 engine_id;
+        u32 active_engine_id = 0;
+        struct fifo_gk20a *f = &g->fifo;
 
-        for (engine_id = 0; engine_id < ENGINE_INVAL_GK20A; engine_id++) {
-                gate_ctrl = gk20a_readl(g, therm_gate_ctrl_r(engine_id));
+        for (engine_id = 0; engine_id < f->num_engines; engine_id++) {
+                active_engine_id = f->active_engines_list[engine_id];
+                gate_ctrl = gk20a_readl(g, therm_gate_ctrl_r(active_engine_id));
                 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_id), gate_ctrl);
+                gk20a_writel(g, therm_gate_ctrl_r(active_engine_id), gate_ctrl);
         }
 
         return 0;