1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index ff81e380..d0c4dec7 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -6703,9 +6703,13 @@ static int gr_gk20a_find_priv_offset_in_ext_buffer(struct gk20a *g,
 
                 gk20a_dbg_info(" gpc = %d tpc = %d",
                                 gpc_num, tpc_num);
+        } else if ((g->ops.gr.is_etpc_addr) &&
+                                g->ops.gr.is_etpc_addr(g, addr)) {
+                        g->ops.gr.get_egpc_etpc_num(g, addr, &gpc_num, &tpc_num);
+                        gpc_base = g->ops.gr.get_egpc_base(g);
         } else {
                 gk20a_dbg(gpu_dbg_fn | gpu_dbg_gpu_dbg,
-                                "extended region has tpc reg only");
+                                "does not exist in extended region");
                 return -EINVAL;
         }