1 files changed, 1 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/falcon/falcon.c b/drivers/gpu/nvgpu/common/falcon/falcon.c
index 375a9cee..b6589bd1 100644
--- a/drivers/gpu/nvgpu/common/falcon/falcon.c
+++ b/drivers/gpu/nvgpu/common/falcon/falcon.c
@@ -142,6 +142,7 @@ void nvgpu_flcn_sw_init(struct gk20a *g, u32 flcn_id)
                 flcn = &g->pmu_flcn;
                 flcn->flcn_id = flcn_id;
                 g->pmu.flcn = &g->pmu_flcn;
+                g->pmu.g = g;
                 break;
         case FALCON_ID_SEC2:
                 flcn = &g->sec2_flcn;