diff --git a/drivers/gpu/nvgpu/gv11b/gr_gv11b.h b/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
index c8912f914..11f5fb47a 100644
--- a/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
+++ b/drivers/gpu/nvgpu/gv11b/gr_gv11b.h
@@ -26,5 +26,8 @@ enum {
 };
 
 void gv11b_init_gr(struct gpu_ops *ops);
+int gr_gv11b_alloc_buffer(struct vm_gk20a *vm, size_t size,
+                        struct mem_desc *mem);
+
 
 #endif
diff --git a/drivers/gpu/nvgpu/gv11b/gv11b.c b/drivers/gpu/nvgpu/gv11b/gv11b.c
index 8e8cf649a..c1809f951 100644
--- a/drivers/gpu/nvgpu/gv11b/gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/gv11b.c
@@ -17,6 +17,7 @@
  */
 
 #include "gk20a/gk20a.h"
+#include "gv11b/gv11b.h"
 
 int gv11b_init_gpu_characteristics(struct gk20a *g)
 {
diff --git a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
index a30e32632..e340c97e5 100644
--- a/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
+++ b/drivers/gpu/nvgpu/gv11b/hal_gv11b.c
@@ -20,6 +20,7 @@
 
 #include "gk20a/gk20a.h"
 
+#include "gv11b/hal_gv11b.h"
 #include "gv11b/gr_gv11b.h"
 #include "gv11b/mc_gv11b.h"
 #include "gv11b/ltc_gv11b.h"