1 files changed, 16 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/ltc.c b/drivers/gpu/nvgpu/common/ltc.c
index 4fbdbe2e..006e2ed2 100644
--- a/drivers/gpu/nvgpu/common/ltc.c
+++ b/drivers/gpu/nvgpu/common/ltc.c
@@ -21,8 +21,10 @@
  */
 
 #include <nvgpu/ltc.h>
+#include <nvgpu/dma.h>
 
 #include "gk20a/gk20a.h"
+#include "gk20a/gr_gk20a.h"
 
 int nvgpu_init_ltc_support(struct gk20a *g)
 {
@@ -46,3 +48,17 @@ void nvgpu_ltc_sync_enabled(struct gk20a *g)
         }
         nvgpu_spinlock_release(&g->ltc_enabled_lock);
 }
+
+int nvgpu_ltc_alloc_cbc(struct gk20a *g, size_t compbit_backing_size)
+{
+        struct gr_gk20a *gr = &g->gr;
+        unsigned long flags = 0;
+
+        if (!nvgpu_iommuable(g))
+                flags = NVGPU_DMA_FORCE_CONTIGUOUS;
+
+        return nvgpu_dma_alloc_flags_sys(g,
+                                         flags,
+                                         compbit_backing_size,
+                                         &gr->compbit_store.mem);
+}