1 files changed, 9 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/pstate/pstate.c b/drivers/gpu/nvgpu/pstate/pstate.c
index a4787f35..e9b9775e 100644
--- a/drivers/gpu/nvgpu/pstate/pstate.c
+++ b/drivers/gpu/nvgpu/pstate/pstate.c
@@ -19,6 +19,7 @@
 #include "pmgr/pmgr.h"
 #include "include/bios.h"
 #include "pstate/pstate.h"
+#include "therm/thrm.h"
 
 static int pstate_sw_setup(struct gk20a *g);
 
@@ -49,6 +50,10 @@ int gk20a_init_pstate_support(struct gk20a *g)
         if (err)
                 return err;
 
+        err = therm_domain_sw_setup(g);
+        if (err)
+                return err;
+
         err = vfe_var_sw_setup(g);
         if (err)
                 return err;
@@ -104,6 +109,10 @@ int gk20a_init_pstate_pmu_support(struct gk20a *g)
                 return err;
         }
 
+        err = therm_domain_pmu_setup(g);
+        if (err)
+                return err;
+
         err = vfe_var_pmu_setup(g);
         if (err)
                 return err;