2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 4ed7251a..aae3072a 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -452,6 +452,9 @@ static int gk20a_pm_prepare_poweroff(struct device *dev)
         ret |= gk20a_mm_suspend(g);
         ret |= gk20a_fifo_suspend(g);
 
+        if (g->ops.pmu.mclk_deinit)
+                g->ops.pmu.mclk_deinit(g);
+
         /* Disable GPCPLL */
         if (g->ops.clk.suspend_clk_support)
                 ret |= g->ops.clk.suspend_clk_support(g);
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 951c8267..e3dad962 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -699,6 +699,7 @@ struct gpu_ops {
                 int (*flcn_populate_bl_dmem_desc)(struct gk20a *g,
                         void *lsfm,     u32 *p_bl_gen_desc_size, u32 falconid);
                 int (*mclk_init)(struct gk20a *g);
+                void (*mclk_deinit)(struct gk20a *g);
                 u32  lspmuwprinitdone;
                 u32  lsfloadedfalconid;
                 bool fecsbootstrapdone;