1 files changed, 4 insertions, 8 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/ltc_gm20b.c b/drivers/gpu/nvgpu/gm20b/ltc_gm20b.c
index 84759192..6ec9aec5 100644
--- a/drivers/gpu/nvgpu/gm20b/ltc_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/ltc_gm20b.c
@@ -24,18 +24,18 @@
 
 #include <trace/events/gk20a.h>
 
-#include "gk20a/gk20a.h"
-
 #include <nvgpu/timers.h>
 #include <nvgpu/enabled.h>
 #include <nvgpu/bug.h>
+#include <nvgpu/ltc.h>
 
 #include <nvgpu/hw/gm20b/hw_mc_gm20b.h>
 #include <nvgpu/hw/gm20b/hw_ltc_gm20b.h>
 #include <nvgpu/hw/gm20b/hw_top_gm20b.h>
 #include <nvgpu/hw/gm20b/hw_pri_ringmaster_gm20b.h>
 
-#include "gk20a/ltc_gk20a.h"
+#include "gk20a/gk20a.h"
+
 #include "ltc_gm20b.h"
 
 int gm20b_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
@@ -92,11 +92,7 @@ int gm20b_ltc_init_comptags(struct gk20a *g, struct gr_gk20a *gr)
         gk20a_dbg_info("max comptag lines : %d",
                 max_comptag_lines);
 
-        if (nvgpu_is_enabled(g, NVGPU_IS_FMODEL))
-                err = gk20a_ltc_alloc_phys_cbc(g, compbit_backing_size);
-        else
-                err = gk20a_ltc_alloc_virt_cbc(g, compbit_backing_size);
-
+        err = nvgpu_ltc_alloc_cbc(g, compbit_backing_size);
         if (err)
                 return err;