2 files changed, 5 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
index 4a22441c..2b98dc50 100644
--- a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
@@ -795,7 +795,7 @@ static void vgpu_remove_gr_support(struct gr_gk20a *gr)
 {
         gk20a_dbg_fn("");
 
-        gk20a_allocator_destroy(&gr->comp_tags);
+        gk20a_comptag_allocator_destroy(&gr->comp_tags);
 
         kfree(gr->gpc_tpc_mask);
         gr->gpc_tpc_mask = NULL;
diff --git a/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c b/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
index 199e880b..76ee5ec9 100644
--- a/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/ltc_vgpu.c
@@ -56,8 +56,10 @@ static int vgpu_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
         if (max_comptag_lines < 2)
                 return -ENXIO;
 
-        __gk20a_allocator_init(&gr->comp_tags, NULL, "comptag",
-                               1, max_comptag_lines - 1, 1, 10, 0); /* length*/
+        err = gk20a_comptag_allocator_init(&gr->comp_tags, max_comptag_lines);
+        if (err)
+                return err;
+
         return 0;
 }