2 files changed, 22 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
index f9fb48b5..a8a031be 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_buddy.c
@@ -1090,6 +1090,19 @@ static u64 gk20a_buddy_alloc_end(struct gk20a_allocator *a)
         return ba->end;
 }
 
+static u64 gk20a_buddy_alloc_space(struct gk20a_allocator *a)
+{
+        struct gk20a_buddy_allocator *ba = a->priv;
+        u64 space;
+
+        alloc_lock(a);
+        space = ba->end - ba->start -
+                (ba->bytes_alloced_real - ba->bytes_freed);
+        alloc_unlock(a);
+
+        return space;
+}
+
 /*
  * Print the buddy allocator top level stats. If you pass @s as NULL then the
  * stats are printed to the kernel log. This lets this code be used for
@@ -1180,6 +1193,7 @@ static const struct gk20a_allocator_ops buddy_ops = {
         .length         = gk20a_buddy_alloc_length,
         .end            = gk20a_buddy_alloc_end,
         .inited         = gk20a_buddy_alloc_inited,
+        .space          = gk20a_buddy_alloc_space,
 
         .fini           = gk20a_buddy_allocator_destroy,
 
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
index 42eb9a14..2e5d46b9 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator_page.c
@@ -57,6 +57,13 @@ static u64 gk20a_page_alloc_end(struct gk20a_allocator *a)
         return gk20a_alloc_end(&va->source_allocator);
 }
 
+static u64 gk20a_page_alloc_space(struct gk20a_allocator *a)
+{
+        struct gk20a_page_allocator *va = a->priv;
+
+        return gk20a_alloc_space(&va->source_allocator);
+}
+
 static int gk20a_page_reserve_co(struct gk20a_allocator *a,
                                  struct gk20a_alloc_carveout *co)
 {
@@ -492,6 +499,7 @@ static const struct gk20a_allocator_ops page_ops = {
         .length         = gk20a_page_alloc_length,
         .end            = gk20a_page_alloc_end,
         .inited         = gk20a_page_alloc_inited,
+        .space          = gk20a_page_alloc_space,
 
         .fini           = gk20a_page_allocator_destroy,