2 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
index 96d21c0a..b913847b 100644
--- a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
@@ -844,7 +844,7 @@ static void vgpu_remove_gr_support(struct gr_gk20a *gr)
 {
         gk20a_dbg_fn("");
 
-        gk20a_comptag_allocator_destroy(&gr->comp_tags);
+        gk20a_comptag_allocator_destroy(gr->g, &gr->comp_tags);
 
         nvgpu_kfree(gr->g, gr->sm_error_states);
         gr->sm_error_states = NULL;
diff --git a/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c b/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
index a6848872..fb9558e2 100644
--- a/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
@@ -50,7 +50,7 @@ int vgpu_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
         if (max_comptag_lines < 2)
                 return -ENXIO;
 
-        err = gk20a_comptag_allocator_init(&gr->comp_tags, max_comptag_lines);
+        err = gk20a_comptag_allocator_init(g, &gr->comp_tags, max_comptag_lines);
         if (err)
                 return err;