1 files changed, 12 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/power_features/cg/cg.c b/drivers/gpu/nvgpu/common/power_features/cg/cg.c
index a538c44b..39796bc8 100644
--- a/drivers/gpu/nvgpu/common/power_features/cg/cg.c
+++ b/drivers/gpu/nvgpu/common/power_features/cg/cg.c
@@ -428,6 +428,9 @@ void nvgpu_cg_init_gr_load_gating_prod(struct gk20a *g)
         if (g->ops.clock_gating.slcg_xbar_load_gating_prod != NULL) {
                 g->ops.clock_gating.slcg_xbar_load_gating_prod(g, true);
         }
+        if (g->ops.clock_gating.slcg_hshub_load_gating_prod != NULL) {
+                g->ops.clock_gating.slcg_hshub_load_gating_prod(g, true);
+        }
 
 check_can_blcg:
         if (!nvgpu_is_enabled(g, NVGPU_GPU_CAN_BLCG)) {
@@ -449,6 +452,9 @@ check_can_blcg:
         if (g->ops.clock_gating.blcg_xbar_load_gating_prod != NULL) {
                 g->ops.clock_gating.blcg_xbar_load_gating_prod(g, true);
         }
+        if (g->ops.clock_gating.blcg_hshub_load_gating_prod != NULL) {
+                g->ops.clock_gating.blcg_hshub_load_gating_prod(g, true);
+        }
 pg_gr_load:
         if (g->ops.clock_gating.pg_gr_load_gating_prod != NULL) {
                 g->ops.clock_gating.pg_gr_load_gating_prod(g, true);
@@ -538,6 +544,9 @@ void nvgpu_cg_blcg_set_blcg_enabled(struct gk20a *g, bool enable)
         if (g->ops.clock_gating.blcg_xbar_load_gating_prod != NULL) {
                 g->ops.clock_gating.blcg_xbar_load_gating_prod(g, enable);
         }
+        if (g->ops.clock_gating.blcg_hshub_load_gating_prod != NULL) {
+                g->ops.clock_gating.blcg_hshub_load_gating_prod(g, enable);
+        }
 
 done:
         nvgpu_mutex_release(&g->cg_pg_lock);
@@ -610,6 +619,9 @@ void nvgpu_cg_slcg_set_slcg_enabled(struct gk20a *g, bool enable)
         if (g->ops.clock_gating.slcg_xbar_load_gating_prod != NULL) {
                 g->ops.clock_gating.slcg_xbar_load_gating_prod(g, enable);
         }
+        if (g->ops.clock_gating.slcg_hshub_load_gating_prod != NULL) {
+                g->ops.clock_gating.slcg_hshub_load_gating_prod(g, enable);
+        }
 
 done:
         nvgpu_mutex_release(&g->cg_pg_lock);