2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
index 5c210519..349548cd 100644
--- a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
+++ b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
@@ -235,6 +235,8 @@ static const struct gpu_ops vgpu_gp10b_ops = {
                 .alloc_global_ctx_buffers = gr_gk20a_alloc_global_ctx_buffers,
                 .map_global_ctx_buffers = gr_gk20a_map_global_ctx_buffers,
                 .commit_global_ctx_buffers = gr_gk20a_commit_global_ctx_buffers,
+                .get_offset_in_gpccs_segment =
+                        gr_gk20a_get_offset_in_gpccs_segment,
         },
         .fb = {
                 .reset = fb_gk20a_reset,
diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
index 2b65c992..d4a113f8 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
@@ -272,6 +272,8 @@ static const struct gpu_ops vgpu_gv11b_ops = {
                 .alloc_global_ctx_buffers = gr_gk20a_alloc_global_ctx_buffers,
                 .map_global_ctx_buffers = gr_gk20a_map_global_ctx_buffers,
                 .commit_global_ctx_buffers = gr_gk20a_commit_global_ctx_buffers,
+                .get_offset_in_gpccs_segment =
+                        gr_gk20a_get_offset_in_gpccs_segment,
         },
         .fb = {
                 .reset = gv11b_fb_reset,