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 91a4964c..47992988 100644
--- a/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
@@ -18,7 +18,6 @@
 #include <dt-bindings/memory/tegra-swgroup.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"
@@ -110,8 +109,9 @@ static int gp10b_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
         if (err)
                 return err;
 
-        __gk20a_allocator_init(&gr->comp_tags, NULL, "comptag",
-                               1, max_comptag_lines - 1, 1, 10, 0);
+        err = gk20a_comptag_allocator_init(&gr->comp_tags, max_comptag_lines);
+        if (err)
+                return err;
 
         gr->comptags_per_cacheline = comptags_per_cacheline;
         gr->slices_per_ltc = slices_per_ltc;