1 files changed, 9 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/pstate/pstate.c b/drivers/gpu/nvgpu/pstate/pstate.c
index cf758023..cca6c445 100644
--- a/drivers/gpu/nvgpu/pstate/pstate.c
+++ b/drivers/gpu/nvgpu/pstate/pstate.c
@@ -79,6 +79,11 @@ int gk20a_init_pstate_support(struct gk20a *g)
                 return err;
 
         err = pmgr_domain_sw_setup(g);
+        if (err)
+                return err;
+
+        err = clk_freq_controller_sw_setup(g);
+
         return err;
 }
 
@@ -141,6 +146,10 @@ int gk20a_init_pstate_pmu_support(struct gk20a *g)
         if (err)
                 return err;
 
+        err = clk_freq_controller_pmu_setup(g);
+        if (err)
+                return err;
+
         err = clk_pmu_vin_load(g);
         if (err)
                 return err;