1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
index 041c7edf..4dbde580 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
@@ -141,11 +141,18 @@ struct gr_ctx_buffer_desc {
         void *priv;
 };
 
+#ifdef CONFIG_ARCH_TEGRA_18x_SOC
+#include "gr_t18x.h"
+#endif
+
 struct gr_ctx_desc {
         struct page **pages;
         u64 iova;
         size_t size;
         u64 gpu_va;
+#ifdef CONFIG_ARCH_TEGRA_18x_SOC
+        struct gr_ctx_desc_t18x t18x;
+#endif
 };
 
 struct compbit_store_desc {