1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/buddy_allocator_priv.h b/drivers/gpu/nvgpu/common/mm/buddy_allocator_priv.h
index 50a11f14..56aaea62 100644
--- a/drivers/gpu/nvgpu/common/mm/buddy_allocator_priv.h
+++ b/drivers/gpu/nvgpu/common/mm/buddy_allocator_priv.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2017, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -20,6 +20,8 @@
 #include <linux/list.h>
 #include <linux/rbtree.h>
 
+#include <nvgpu/kmem.h>
+
 struct nvgpu_allocator;
 struct vm_gk20a;
 
@@ -126,6 +128,8 @@ struct nvgpu_buddy_allocator {
 
         struct list_head co_list;
 
+        struct nvgpu_kmem_cache *buddy_cache;
+
         /*
          * Impose an upper bound on the maximum order.
          */