1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 78e33a49..96711c0c 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -609,8 +609,9 @@ static void gk20a_remove_support(struct platform_device *dev)
 {
         struct gk20a *g = get_gk20a(dev);
 
-        /* pmu support should already be removed when driver turns off
-           gpu power rail in prepapre_poweroff */
+        if (g->pmu.remove_support)
+                g->pmu.remove_support(&g->pmu);
+
         if (g->gk20a_cdev.gk20a_cooling_dev)
                 thermal_cooling_device_unregister(g->gk20a_cdev.gk20a_cooling_dev);