gpu: nvgpu: PG statistics update

- PG statistics read support for multiple engines

JIRA DNVGPU-71

Change-Id: I2dc3aad243300d21dc3d20a54a5e4736977e071b
Signed-off-by: Mahantesh Kumbar <mkumbar@nvidia.com>
Reviewed-on: http://git-master/r/1250507
(cherry picked from commit 985cb3be1d6d990bc6651e417d9e6ba9bfe306e0)
Reviewed-on: http://git-master/r/1270991
Reviewed-by: Automatic_Commit_Validation_User
GVS: Gerrit_Virtual_Submit
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

1 files changed, 3 insertions, 2 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;