2 files changed, 4 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
index 9274990a..12337934 100644
--- a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
@@ -250,13 +250,14 @@ int gp10b_pg_gr_init(struct gk20a *g, u32 pg_engine_id)
         return 0;
 }
 
-void gp10b_pmu_elpg_statistics(struct gk20a *g,
+void gp10b_pmu_elpg_statistics(struct gk20a *g, u32 pg_engine_id,
                 u32 *ingating_time, u32 *ungating_time, u32 *gating_cnt)
 {
         struct pmu_gk20a *pmu = &g->pmu;
         struct pmu_pg_stats_v1 stats;
 
-        pmu_copy_from_dmem(pmu, pmu->stat_dmem_offset,
+        pmu_copy_from_dmem(pmu,
+                pmu->stat_dmem_offset[pg_engine_id],
                 (u8 *)&stats, sizeof(struct pmu_pg_stats_v1), 0);
 
         *ingating_time = stats.total_sleep_timeus;
diff --git a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.h b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.h
index ee0158ec..c9ac9d41 100644
--- a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.h
@@ -20,7 +20,7 @@ void gp10b_init_pmu_ops(struct gpu_ops *gops);
 int gp10b_load_falcon_ucode(struct gk20a *g, u32 falconidmask);
 int gp10b_pg_gr_init(struct gk20a *g, u32 pg_engine_id);
 void gp10b_write_dmatrfbase(struct gk20a *g, u32 addr);
-void gp10b_pmu_elpg_statistics(struct gk20a *g,
+void gp10b_pmu_elpg_statistics(struct gk20a *g, u32 pg_engine_id,
                 u32 *ingating_time, u32 *ungating_time, u32 *gating_cnt);
 
 #endif /*__PMU_GP10B_H_*/