1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/common/mm/bitmap_allocator_priv.h b/drivers/gpu/nvgpu/common/mm/bitmap_allocator_priv.h
index 9802b9db..95780202 100644
--- a/drivers/gpu/nvgpu/common/mm/bitmap_allocator_priv.h
+++ b/drivers/gpu/nvgpu/common/mm/bitmap_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,
@@ -19,6 +19,8 @@
 
 #include <linux/rbtree.h>
 
+#include <nvgpu/kmem.h>
+
 struct nvgpu_allocator;
 
 struct nvgpu_bitmap_allocator {
@@ -43,6 +45,8 @@ struct nvgpu_bitmap_allocator {
         unsigned long *bitmap;          /* The actual bitmap! */
         struct rb_root allocs;          /* Tree of outstanding allocations. */
 
+        struct nvgpu_kmem_cache *meta_data_cache;
+
         u64 flags;
 
         bool inited;