1 files changed, 30 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/mm.c b/drivers/gpu/nvgpu/common/mm/mm.c
index 88e3baa8..6c8f2cdf 100644
--- a/drivers/gpu/nvgpu/common/mm/mm.c
+++ b/drivers/gpu/nvgpu/common/mm/mm.c
@@ -198,6 +198,10 @@ static void nvgpu_remove_mm_support(struct mm_gk20a *mm)
         nvgpu_semaphore_sea_destroy(g);
         nvgpu_vidmem_destroy(g);
         nvgpu_pd_cache_fini(g);
+
+        if (g->ops.fifo.deinit_pdb_cache_war) {
+                g->ops.fifo.deinit_pdb_cache_war(g);
+        }
 }
 
 /* pmu vm, share channel_vm interfaces */
@@ -501,6 +505,27 @@ static int nvgpu_init_mm_setup_sw(struct gk20a *g)
         return 0;
 }
 
+static int nvgpu_init_mm_pdb_cache_war(struct gk20a *g)
+{
+        int err;
+
+        if (g->ops.fifo.init_pdb_cache_war) {
+                err = g->ops.fifo.init_pdb_cache_war(g);
+                if (err) {
+                        return err;
+                }
+        }
+
+        if (g->ops.fb.apply_pdb_cache_war) {
+                err = g->ops.fb.apply_pdb_cache_war(g);
+                if (err) {
+                        return err;
+                }
+        }
+
+        return 0;
+}
+
 int nvgpu_init_mm_support(struct gk20a *g)
 {
         u32 err;
@@ -510,6 +535,11 @@ int nvgpu_init_mm_support(struct gk20a *g)
                 return err;
         }
 
+        err = nvgpu_init_mm_pdb_cache_war(g);
+        if (err) {
+                return err;
+        }
+
         err = nvgpu_init_mm_setup_sw(g);
         if (err) {
                 return err;