1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gv11b/mm_gv11b.c b/drivers/gpu/nvgpu/gv11b/mm_gv11b.c
index e1d90829..0cf08d4a 100644
--- a/drivers/gpu/nvgpu/gv11b/mm_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/mm_gv11b.c
@@ -249,7 +249,11 @@ void gv11b_mm_l2_flush(struct gk20a *g, bool invalidate)
 
         g->ops.mm.fb_flush(g);
         gk20a_mm_l2_flush(g, invalidate);
-        g->ops.mm.fb_flush(g);
+        if (g->ops.bus.bar1_bind)
+                g->ops.fb.tlb_invalidate(g,
+                                g->mm.bar1.vm->pdb.mem);
+        else
+                g->ops.mm.fb_flush(g);
 }
 
 /*