2 files changed, 3 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index c2a8d6ce..e3b37747 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -244,7 +244,8 @@ struct gpu_ops {
                         u32 addr, int *addr_type,
                         u32 *gpc_num, u32 *tpc_num, u32 *broadcast_flags);
                 void (*egpc_etpc_priv_addr_table)(struct gk20a *g, u32 addr,
-                        u32 gpc, u32 broadcast_flags, u32 *priv_addr_table,
+                        u32 gpc, u32 tpc, u32 broadcast_flags,
+                        u32 *priv_addr_table,
                         u32 *priv_addr_table_index);
                 bool (*is_tpc_addr)(struct gk20a *g, u32 addr);
                 bool (*is_egpc_addr)(struct gk20a *g, u32 addr);
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index e4344b9f..96bc72af 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -6421,7 +6421,7 @@ int gr_gk20a_create_priv_addr_table(struct gk20a *g,
                         (addr_type == CTXSW_ADDR_TYPE_ETPC)) &&
                                 g->ops.gr.egpc_etpc_priv_addr_table) {
                 gk20a_dbg(gpu_dbg_gpu_dbg, "addr_type : EGPC/ETPC");
-                g->ops.gr.egpc_etpc_priv_addr_table(g, addr, gpc_num,
+                g->ops.gr.egpc_etpc_priv_addr_table(g, addr, gpc_num, tpc_num,
                                 broadcast_flags, priv_addr_table, &t);
         } else if (broadcast_flags & PRI_BROADCAST_FLAGS_LTSS) {
                 g->ops.gr.split_lts_broadcast_addr(g, addr,