1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 70eb0a7e..606dbc1f 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -696,6 +696,9 @@ void gk20a_remove_support(struct device *dev)
         if (g->gr.remove_support)
                 g->gr.remove_support(&g->gr);
 
+        if (g->mm.remove_ce_support)
+                g->mm.remove_ce_support(&g->mm);
+
         if (g->fifo.remove_support)
                 g->fifo.remove_support(&g->fifo);