1 files changed, 33 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/mm.c b/drivers/gpu/nvgpu/common/mm/mm.c
index 54f621ae..988b1e5c 100644
--- a/drivers/gpu/nvgpu/common/mm/mm.c
+++ b/drivers/gpu/nvgpu/common/mm/mm.c
@@ -173,6 +173,9 @@ static void nvgpu_remove_mm_support(struct mm_gk20a *mm)
 {
         struct gk20a *g = gk20a_from_mm(mm);
 
+        nvgpu_dma_free(g, &mm->mmu_wr_mem);
+        nvgpu_dma_free(g, &mm->mmu_rd_mem);
+
         if (g->ops.mm.fault_info_mem_destroy) {
                 g->ops.mm.fault_info_mem_destroy(g);
         }
@@ -294,6 +297,32 @@ static int nvgpu_init_ce_vm(struct mm_gk20a *mm)
         return 0;
 }
 
+static int nvgpu_init_mmu_debug(struct mm_gk20a *mm)
+{
+        struct gk20a *g = gk20a_from_mm(mm);
+        int err;
+
+        if (!nvgpu_mem_is_valid(&mm->mmu_wr_mem)) {
+                err = nvgpu_dma_alloc_sys(g, SZ_4K, &mm->mmu_wr_mem);
+                if (err) {
+                        goto err;
+                }
+        }
+
+        if (!nvgpu_mem_is_valid(&mm->mmu_rd_mem)) {
+                err = nvgpu_dma_alloc_sys(g, SZ_4K, &mm->mmu_rd_mem);
+                if (err) {
+                        goto err_free_wr_mem;
+                }
+        }
+        return 0;
+
+ err_free_wr_mem:
+        nvgpu_dma_free(g, &mm->mmu_wr_mem);
+ err:
+        return -ENOMEM;
+}
+
 void nvgpu_init_mm_ce_context(struct gk20a *g)
 {
 #if defined(CONFIG_GK20A_VIDMEM)
@@ -459,6 +488,10 @@ static int nvgpu_init_mm_setup_sw(struct gk20a *g)
                 return err;
         }
 
+        err = nvgpu_init_mmu_debug(mm);
+        if (err)
+                return err;
+
         mm->remove_support = nvgpu_remove_mm_support;
         mm->remove_ce_support = nvgpu_remove_mm_ce_support;