1 files changed, 5 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
index 184c1f71..d7503948 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
@@ -391,9 +391,11 @@ struct mm_gk20a {
         bool force_pramin; /* via debugfs */
 #endif
 
-        size_t vidmem_size;
-        struct device vidmem_dev;
-        u32 ce_vidmem_ctx_id;
+        struct {
+                size_t size;
+                struct gk20a_allocator allocator;
+                u32 ce_ctx_id;
+        } vidmem;
 };
 
 int gk20a_mm_init(struct mm_gk20a *mm);