gpu: nvgpu: support etpc addresses in ext buffer

Bug 1960226

Change-Id: Ifdcdff803e96c59b2794170349ec4e2c00406706
Signed-off-by: Seema Khowala <seemaj@nvidia.com>
Reviewed-on: https://git-master.nvidia.com/r/1522445
GVS: Gerrit_Virtual_Submit
Reviewed-by: Vijayakumar Subbu <vsubbu@nvidia.com>

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;
         }