1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c b/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
index 3c809eaf..68f4eafa 100644
--- a/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
@@ -16,6 +16,7 @@
 #include <linux/types.h>
 
 #include "gk20a/gk20a.h"
+#include "gk20a/gk20a_allocator.h"
 #include "gm20b/ltc_gm20b.h"
 #include "hw_proj_gp10b.h"
 #include "hw_mc_gp10b.h"
@@ -111,9 +112,8 @@ static int gp10b_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
         if (err)
                 return err;
 
-        gk20a_allocator_init(&gr->comp_tags, "comptag",
-                              1, /* start */
-                              max_comptag_lines - 1); /* length*/
+        __gk20a_allocator_init(&gr->comp_tags, NULL, "comptag",
+                               1, max_comptag_lines - 1, 1, 10, 0);
 
         gr->comptags_per_cacheline = comptags_per_cacheline;
         gr->slices_per_ltc = slices_per_ltc;