1 files changed, 11 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/pstate/pstate.c b/drivers/gpu/nvgpu/pstate/pstate.c
index 791832cb..4ce7268e 100644
--- a/drivers/gpu/nvgpu/pstate/pstate.c
+++ b/drivers/gpu/nvgpu/pstate/pstate.c
@@ -26,6 +26,9 @@ static int pstate_sw_setup(struct gk20a *g);
 
 void gk20a_deinit_pstate_support(struct gk20a *g)
 {
+        if (g->ops.clk.mclk_deinit)
+                g->ops.clk.mclk_deinit(g);
+
         nvgpu_mutex_destroy(&g->perf_pmu.pstatesobjs.pstate_mutex);
 }
 
@@ -104,6 +107,14 @@ int gk20a_init_pstate_pmu_support(struct gk20a *g)
 
         gk20a_dbg_fn("");
 
+        if (g->ops.clk.mclk_init) {
+                err = g->ops.clk.mclk_init(g);
+                if (err) {
+                        nvgpu_err(g, "failed to set mclk");
+                        /* Indicate error and continue */
+                }
+        }
+
         err = volt_rail_pmu_setup(g);
         if (err)
                 return err;