2 files changed, 10 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
index f8e55788..15d5b732 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.c
@@ -59,6 +59,14 @@ u64 gk20a_alloc_end(struct gk20a_allocator *a)
         return 0;
 }
 
+u64 gk20a_alloc_space(struct gk20a_allocator *a)
+{
+        if (a->ops->space)
+                return a->ops->space(a);
+
+        return 0;
+}
+
 u64 gk20a_alloc(struct gk20a_allocator *a, u64 len)
 {
         return a->ops->alloc(a, len);
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.h b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.h
index 40388ef8..dc065ffa 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a_allocator.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a_allocator.h
@@ -61,6 +61,7 @@ struct gk20a_allocator_ops {
         u64  (*length)(struct gk20a_allocator *allocator);
         u64  (*end)(struct gk20a_allocator *allocator);
         int  (*inited)(struct gk20a_allocator *allocator);
+        u64  (*space)(struct gk20a_allocator *allocator);
 
         /* Destructor. */
         void (*fini)(struct gk20a_allocator *allocator);
@@ -223,6 +224,7 @@ u64  gk20a_alloc_base(struct gk20a_allocator *a);
 u64  gk20a_alloc_length(struct gk20a_allocator *a);
 u64  gk20a_alloc_end(struct gk20a_allocator *a);
 u64  gk20a_alloc_initialized(struct gk20a_allocator *a);
+u64  gk20a_alloc_space(struct gk20a_allocator *a);
 
 void gk20a_alloc_destroy(struct gk20a_allocator *allocator);