1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
index b2cca072..1a9bee5f 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
@@ -376,6 +376,11 @@ struct mm_gk20a {
 #ifdef CONFIG_ARCH_TEGRA_18x_SOC
         struct mem_desc bar2_desc;
 #endif
+        /*
+         * Separate function to cleanup the CE since it requires a channel to
+         * be closed which must happen before fifo cleanup.
+         */
+        void (*remove_ce_support)(struct mm_gk20a *mm);
         void (*remove_support)(struct mm_gk20a *mm);
         bool sw_ready;
         int physical_bits;