gpu: nvgpu: Do not clear PMU state on rail gate

When rail gating, we cleared all PMU status. Clear only the relevant
fields.

Change-Id: I5b4e8d74339aae6f1c6b945f45b8378bb563e8be
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/406843

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);