1 files changed, 10 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
index 2c8eb16d..db99535a 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.h
@@ -27,6 +27,7 @@
 #include <asm/cacheflush.h>
 
 #include <nvgpu/allocator.h>
+#include <nvgpu/list.h>
 
 #ifdef CONFIG_ARM64
 #define outer_flush_range(a, b)
@@ -76,11 +77,18 @@ struct mem_desc {
         bool fixed; /* vidmem only */
         bool user_mem; /* vidmem only */
         struct nvgpu_allocator *allocator; /* vidmem only */
-        struct list_head clear_list_entry; /* vidmem only */
+        struct nvgpu_list_node clear_list_entry; /* vidmem only */
         bool skip_wmb;
         unsigned long flags;
 };
 
+static inline struct mem_desc *
+mem_desc_from_clear_list_entry(struct nvgpu_list_node *node)
+{
+        return (struct mem_desc *)
+                ((uintptr_t)node - offsetof(struct mem_desc, clear_list_entry));
+};
+
 struct mem_desc_sub {
         u32 offset;
         u32 size;
@@ -416,7 +424,7 @@ struct mm_gk20a {
                 volatile bool cleared;
                 struct nvgpu_mutex first_clear_mutex;
 
-                struct list_head clear_list_head;
+                struct nvgpu_list_node clear_list_head;
                 struct nvgpu_mutex clear_list_mutex;
 
                 struct work_struct clear_mem_worker;