1 files changed, 15 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
index d8af5d7c..93247411 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
@@ -525,6 +525,12 @@ struct pmu_payload {
         } in, out;
 };
 
+struct pmu_surface {
+        struct mem_desc vidmem_desc;
+        struct mem_desc sysmem_desc;
+        struct flcn_mem_desc_v0 params;
+};
+
 typedef void (*pmu_callback)(struct gk20a *, struct pmu_msg *, void *, u32,
         u32);
 
@@ -539,12 +545,14 @@ struct pmu_sequence {
                 struct pmu_allocation_v2 in_v2;
                 struct pmu_allocation_v3 in_v3;
         };
+        struct mem_desc *in_mem;
         union {
                 struct pmu_allocation_v0 out_v0;
                 struct pmu_allocation_v1 out_v1;
                 struct pmu_allocation_v2 out_v2;
                 struct pmu_allocation_v3 out_v3;
         };
+        struct mem_desc *out_mem;
         u8 *out_payload;
         pmu_callback callback;
         void* cb_params;
@@ -797,4 +805,11 @@ int gk20a_pmu_reset(struct gk20a *g);
 int pmu_idle(struct pmu_gk20a *pmu);
 int pmu_enable_hw(struct pmu_gk20a *pmu, bool enable);
 
+void gk20a_pmu_surface_free(struct gk20a *g, struct mem_desc *mem);
+void gk20a_pmu_surface_describe(struct gk20a *g, struct mem_desc *mem,
+                struct flcn_mem_desc_v0 *fb);
+int gk20a_pmu_vidmem_surface_alloc(struct gk20a *g, struct mem_desc *mem,
+                u32 size);
+int gk20a_pmu_sysmem_surface_alloc(struct gk20a *g, struct mem_desc *mem,
+                u32 size);
 #endif /*__PMU_GK20A_H__*/